Update app.py

app.py

@@ -120,9 +120,9 @@ if uploaded_file:
         ]
     }
 
-    predefined_q = st.selectbox("Choose a predefined question (optional)", ["None"] + predefined_qas[role])
-    manual_q = st.text_input("Or type your own question below:")
-    question = manual_q if manual_q.strip() else (predefined_q if predefined_q != "None" else "")
+    predefined_q = st.selectbox("Choose a predefined question", predefined_qas[role])
+    manual_q = st.text_input("Or type your own question below (optional):")
+    question = manual_q.strip() if manual_q else predefined_q
 
     if question:
         with st.spinner("Generating insights..."):
@@ -147,7 +147,8 @@ QUESTION: {question}
 Provide a short, focused response in your role.
 """
             prompt = f"<s>[INST] {context} [/INST]"
-            output = nlp(prompt, max_new_tokens=250, do_sample=True, temperature=0.5)[0]['generated_text']
+            result = nlp(prompt, max_new_tokens=250, do_sample=True, temperature=0.5)
+            output = result[0]['generated_text']
 
             if '[/INST]' in output:
                 answer = output.split('[/INST]')[-1].strip()
@@ -159,4 +160,3 @@ Provide a short, focused response in your role.
 
 else:
     st.info("📂 Please upload a factory CSV data file to begin analysis.")
-