Update app.py

app.py

@@ -1,9 +1,9 @@
 import streamlit as st
 from openai import OpenAI
+from config import config
 import requests
 from prompts import SYSTEM_PROMPT
 import time
-import os
 
 
 def setup_page():
@@ -20,13 +20,12 @@ def initialize_session_state():
     """Initialize Streamlit session state variables."""
     if "messages" not in st.session_state:
         st.session_state.messages = []
-    if "ollama_server" not in st.session_state:
-        st.session_state.ollama_server = "http://127.0.0.1:11434"
 
 
 def get_ollama_client():
     """Initialize the OpenAI client for Ollama."""
-    base_url = f"{st.session_state.ollama_server}/v1/"
+    ollama_server = config.OLLAMA_SERVER
+    base_url = f"{ollama_server}/v1/"
     return OpenAI(base_url=base_url, api_key="ollama")
 
 
@@ -44,9 +43,9 @@ def chat_completion(client, messages, model):
     return response.choices[0].message.content
 
 
-def get_models():
+def get_models(ollama_server):
     """Fetch available models from the Ollama server."""
-    url = f"{st.session_state.ollama_server}/api/tags"
+    url = f"{ollama_server}/api/tags"
     try:
         response = requests.get(url)
         response.raise_for_status()
@@ -97,14 +96,14 @@ def main():
     st.title("Ollama Chatbot")
     st.sidebar.title("Settings")
 
-    # Display the hardcoded Ollama server URL
-    st.sidebar.text(f"Ollama Server: {st.session_state.ollama_server}")
+    # Ollama Server configuration
+    ollama_server = st.sidebar.text_input("Ollama Server", value=config.OLLAMA_SERVER)
 
     # Initialize the OpenAI client
     client = get_ollama_client()
 
     # Fetch available models
-    models = get_models()
+    models = get_models(ollama_server)
     if not models:
         st.error("No models available. Please check the server connection.")
         return
@@ -117,7 +116,8 @@ def main():
     update = st.sidebar.button("Update settings", use_container_width=True)
 
     if update:
-        # Update the system prompt
+        # Update configuration and system prompt
+        config.OLLAMA_SERVER = ollama_server
         SYSTEM_PROMPT = update_prompt
         if len(st.session_state.messages) == 0:
             st.session_state.messages.insert(0, create_message("system", SYSTEM_PROMPT))