Update app.py

app.py

@@ -167,20 +167,20 @@ def predict(user_input,company):
             "traceback": traceback.format_exc()
         }
 
-# Log safely
-with scheduler.lock:
-    with log_file.open("a") as f:
-        f.write(json.dumps(
-            {
-                'user_input': user_input,
-                'retrieved_context': context_for_query,
-                'model_response': prediction
-            },
-            indent=2
-        ))
-        f.write("\n")
-
-return prediction
+    # Log safely
+    with scheduler.lock:
+        with log_file.open("a") as f:
+            f.write(json.dumps(
+                {
+                    'user_input': user_input,
+                    'retrieved_context': context_for_query,
+                    'model_response': prediction
+                },
+                indent=2
+            ))
+            f.write("\n")
+    
+    return prediction
 
 examples = [
     ["What are the AWS policies and frameworks regarding AI ethics, governance, and responsible AI use as detailed in their 10-K reports?", "AWS"],