Update app.py

app.py

@@ -74,16 +74,19 @@ class AuginiDarkTheme(Base):
         )
 
 class AuginiChat:
-    def __init__(self):
+    def __init__(self, model: str, temperature: float = 0.7):
         self.df: Optional[pd.DataFrame] = None
+        self.model = model
+        self.temperature = temperature
         # Initialize Augini with the API key directly
         self.augini = Augini(
             api_key=os.environ.get('OPENROUTER_TOKEN'),
             use_openrouter=True,
-            model='gpt-4o-mini',
-            temperature=0.7,
-            max_tokens=1500, 
+            model=self.model,
+            temperature=self.temperature,
+            max_tokens=1500,
         )
+
     
     def upload_file(self, file) -> str:
         """Handle file upload and return preview"""
@@ -127,9 +130,20 @@ class AuginiChat:
             error_msg = f"❌ Error processing message: {str(e)}"
             return "", history + [(message, error_msg)]
 
+    def update_model_settings(self, model_name: str, temperature: float) -> None:
+        """Update the model settings and reinitialize Augini."""
+        self.model = model_name
+        self.temperature = temperature
+        self.augini = Augini(
+            api_key=os.environ.get('OPENROUTER_TOKEN'),
+            use_openrouter=True,
+            model=self.model,
+            temperature=self.temperature,
+        )
+
 def create_app():
-    # Initialize the chat handler
-    chat_handler = AuginiChat()
+    # Initialize the chat handler with default settings
+    chat_handler = AuginiChat(model='openai/gpt-4o-mini', temperature=0.7)
     
     # JavaScript to force dark theme - added to head
     dark_mode_script = """
@@ -147,6 +161,14 @@ def create_app():
         setDarkTheme();
     </script>
     """
+
+    available_models = [
+        "mistralai/mistral-nemo",
+        "meta-llama/llama-3.3-70b-instruct",
+        "qwen/qwen-2.5-72b-instruct",
+        "openai/gpt-4o-mini",
+        "meta-llama/llama-3.2-3b-instruct",
+    ]
     
     # Create the Gradio interface with dark theme script in head
     with gr.Blocks(head=dark_mode_script) as app:
@@ -159,6 +181,34 @@ def create_app():
         > 💡 **Tip**: Ask questions about patterns, relationships, or any aspect of your data. **augini** will provide detailed, evidence-based answers.
         """, elem_classes=["center-content"])
         
+        with gr.Accordion("⚙️ Model Settings", open=False):
+            model_dropdown = gr.Dropdown(
+                label="Select Model",
+                choices=available_models,
+                value="openai/gpt-4o-mini"
+            )
+            temperature_slider = gr.Slider(
+                label="Temperature",
+                minimum=0.0,
+                maximum=1.0,
+                value=0.7,
+                step=0.05
+            )
+
+            def update_settings(model_name, temperature):
+                chat_handler.update_model_settings(model_name, temperature)
+                return f"Model settings updated: {model_name}, Temperature: {temperature}"
+
+            update_button = gr.Button("Update Model Settings")
+            update_status = gr.Textbox(label="Update Status", interactive=False)
+
+            update_button.click(
+                update_settings,
+                inputs=[model_dropdown, temperature_slider],
+                outputs=[update_status]
+            )
+
+        
         with gr.Row(elem_classes=["container"]):
             # Left sidebar for file upload
             with gr.Column(scale=1, elem_classes=["sidebar"]):
@@ -269,4 +319,4 @@ def create_app():
 
 if __name__ == "__main__":
     app = create_app()
-    app.launch(share=True) 
+    app.launch(share=True)