Update app.py

app.py

@@ -55,7 +55,13 @@ class ResearchOrchestrator:
             formatted_output = self.formatter.format_response(cited_analysis, search_results)
             logging.info("Response formatted successfully")
             
-            yield formatted_output
+            # Add completion notification
+            if len(search_results) >= 3:
+                completion_message = "\n\n---\n[ANALYSIS COMPLETE] ✅ Research finished with sufficient sources."
+            else:
+                completion_message = "\n\n---\n[RECOMMEND FURTHER ANALYSIS] ⚠️ Limited sources found. Consider refining your query."
+            
+            yield formatted_output + completion_message
             
         except Exception as e:
             error_msg = f"❌ An error occurred: {str(e)}"
@@ -144,4 +150,5 @@ with gr.Blocks(title="Research Assistant") as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()
+