Update app.py

app.py

@@ -22,7 +22,7 @@ def get_available_models() -> List[str]:
         response = requests.get("https://openrouter.ai/api/v1/models", headers=headers)
         response.raise_for_status()
         models = response.json()
-        return [model["id"] for model in models]
+        return [model["id"] for model in models.data]
     except Exception as e:
         print(f"Error fetching models: {e}")
         # Fallback to a basic list of known models
@@ -118,30 +118,32 @@ IMPORTANT: Your response must be a valid JSON object with the following structur
         step_md += "---\n\n"
         
         markdown_output += step_md
-        yield markdown_output  # Update the output incrementally
+        #yield markdown_output  # Update the output incrementally
         
         messages.append({"role": "assistant", "content": json.dumps(step_data.model_dump())})
         
         if step_data.next_action == 'final_answer' and step_count < 15:
             messages.append({"role": "user", "content": "Please continue your analysis with at least 5 more steps before providing the final answer."})
-        elif step_data.next_action == 'final_answer':
+        elif step_data.next_action == 'final_answer':Please
             break
         elif step_data.next_action == 'reflect' or step_count % 3 == 0:
-            messages.append({"role": "user", "content": "Please perform a detailed self-reflection on your reasoning so far."})
+            messages.append({"role": "user", "content": " perform a detailed self-reflection on your reasoning so far."})
         else:
             messages.append({"role": "user", "content": "Please continue with the next step in your analysis."})
         
         step_count += 1
+        yield messages
 
     # Generate final answer
     final_data = make_api_call(model, system_prompt, messages, 750, is_final_answer=True)
+    yield messages
+
+    #final_md = f"### Final Answer\n\n"
+    #final_md += f"{final_data.content}\n\n"
+    #final_md += f"**Confidence:** {final_data.confidence:.2f}\n\n"
     
-    final_md = f"### Final Answer\n\n"
-    final_md += f"{final_data.content}\n\n"
-    final_md += f"**Confidence:** {final_data.confidence:.2f}\n\n"
-    
-    markdown_output += final_md
-    yield markdown_output
+    #markdown_output += final_md
+    #yield markdown_output
 
 def create_interface():
     # Check for API key
@@ -161,18 +163,21 @@ def create_interface():
                     value=available_models[0],
                     label="Select Model"
                 )
+                chatbot = gr.Chatbot()
+
                 query_input = gr.Textbox(
+                    lines=5
                     label="Enter your query:",
                     placeholder="e.g., What are the potential long-term effects of climate change on global agriculture?"
                 )
                 submit_btn = gr.Button("Generate Response")
             
-            output_box = gr.Markdown(label="Response")
+            #output_box = gr.Markdown(label="Response")
         
         submit_btn.click(
             fn=generate_response,
             inputs=[query_input, model_dropdown],
-            outputs=output_box
+            outputs=chatbot
         )
     
     return interface