Update pages/1_Simple-Gemini.py

pages/1_Simple-Gemini.py

@@ -168,18 +168,18 @@ def main():
     )
 
 
-        workflow.add_edge('action', 'agent')
-        workflow.add_edge('final', END)
-        # uncomment if you want to always calls a certain tool first
-        # workflow.add_edge('first_agent', 'action')
-        app = workflow.compile()
-
-        inputs = {"input": input_text, "chat_history": [], "return_direct": False}
-        results = []
-        for s in app.stream(inputs):
-            result = list(s.values())[0]
-            results.append(result)
-            st.write(result)  # Display each step's output
+    workflow.add_edge('action', 'agent')
+    workflow.add_edge('final', END)
+    # uncomment if you want to always calls a certain tool first
+    # workflow.add_edge('first_agent', 'action')
+    app = workflow.compile()
+
+    inputs = {"input": input_text, "chat_history": [], "return_direct": False}
+    results = []
+    for s in app.stream(inputs):
+        result = list(s.values())[0]
+        results.append(result)
+        st.write(result)  # Display each step's output
 
 #result = app.invoke({"input": input_text, "chat_history": [], "return_direct": False})