Update app.py

app.py

@@ -3,63 +3,52 @@ import os
 from mistralai import Mistral
 
 api_key = os.getenv("MISTRAL_API_KEY")
+
+if not api_key:
+    raise ValueError("MISTRAL_API_KEY not found in Secrets.")
+
 client = Mistral(api_key=api_key)
 
 def analyze_fault(log_input):
     if not log_input.strip():
         return "Please provide a factory log."
 
-    prompt = f"""
+    try:
+        prompt = f"""
 You are an industrial automation expert working in an ABB Smart Factory.
 
 Use Kepner-Tregoe (KT) structured problem analysis methodology.
 
-Follow strictly:
-1. Problem Description (What, Where, When, Extent)
-2. What IS analysis
-3. What IS NOT analysis
-4. Possible Causes
-5. Most Probable Root Cause
-6. Corrective Action
-7. Preventive Action
-8. Impact Assessment
-9. Severity Level (Low/Medium/High/Critical)
-10. Confidence Score (0-100%)
+Return output in structured JSON format only.
 
 Factory Log:
 ---
 {log_input}
 ---
-
-Return output in structured JSON format only.
 """
 
-    response = client.chat.complete(
-        model="mistral-large-latest",
-        messages=[{"role": "user", "content": prompt}],
-        temperature=0.2
-    )
+        response = client.chat.complete(
+            model="mistral-large-latest",
+            messages=[{"role": "user", "content": prompt}],
+            temperature=0.2,
+            max_tokens=800
+        )
 
-    return response.choices[0].message.content
+        return response.choices[0].message.content
+
+    except Exception as e:
+        return f"Error occurred: {str(e)}"
 
 
 with gr.Blocks(title="ABB Smart Factory – KT Fault Intelligence") as demo:
     gr.Markdown("## 🏭 ABB Smart Factory – AI KT Fault Intelligence System")
-    gr.Markdown("Paste Smart Factory Log below and analyze using structured KT methodology.")
-
-    log_input = gr.Textbox(
-        label="Factory Log Input",
-        lines=12,
-        placeholder="Paste factory PLC or robotic log here..."
-    )
 
-    output = gr.Textbox(
-        label="AI KT Analysis Output",
-        lines=20
-    )
+    log_input = gr.Textbox(label="Factory Log Input", lines=12)
+    output = gr.Textbox(label="AI KT Analysis Output", lines=20)
 
     analyze_button = gr.Button("Analyze Fault")
-
     analyze_button.click(analyze_fault, inputs=log_input, outputs=output)
 
+
+
 demo.launch(server_name="0.0.0.0", server_port=7860)