Dockerfile_updated

Author: RobuRishabh

Dockerfile

@@ -0,0 +1,30 @@
+# Use an official Python image as a base
+FROM python:3.10
+
+# Set the working directory in the container
+WORKDIR /app
+
+# Copy all necessary files
+COPY . .
+
+# Install system dependencies
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    build-essential \
+    && rm -rf /var/lib/apt/lists/*
+
+# Install dependencies
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Expose the port Streamlit runs on
+EXPOSE 8501
+
+# Set environment variables
+# ENV OLLAMA_URL=http://localhost:11434
+ENV OLLAMA_URL=http://10.0.0.210:11434
+# ENV OLLAMA_URL=http://host.docker.internal:11434
+ENV OLLAMA_MODEL=llama2:latest
+
+
+# Command to run the Streamlit app
+CMD ["streamlit", "run", "app.py"]
+# CMD ["streamlit", "run", "app.py", "--server.port=8501", "--server.address=0.0.0.0"]

mock_database.py

@@ -1,12 +1,19 @@
+import os
 import requests
 
 def check_ollama_availability():
     """Check if the Ollama server is available."""
+    ollama_url = os.getenv("OLLAMA_URL", "http://localhost:11434")
     try:
-        response = requests.get("http://localhost:11434/api/tags", timeout=3)
-        return response.status_code == 200
-    except requests.RequestException:
-        print("⚠️ Ollama server is not available.")
+        response = requests.get(f"{ollama_url}/api/tags", timeout=3)
+        if response.status_code == 200:
+            print(f"🟢 Ollama server is available at {ollama_url}.")
+            return True
+        else:
+            print(f"⚠️ Ollama server returned status {response.status_code} at {ollama_url}.")
+            return False
+    except requests.RequestException as e:
+        print(f"⚠️ Ollama server is not available at {ollama_url}: {str(e)}")
         return False
 
 

ollama_api.py

@@ -19,7 +19,7 @@
 def initialize_ollama():
     """Initialize the Ollama LLM model safely."""
     try:
-        ollama_llm = OllamaLLM(model=OLLAMA_MODEL)
+        ollama_llm = OllamaLLM(model=OLLAMA_MODEL, base_url=OLLAMA_URL)
         print(f"🟢 Successfully initialized Ollama LLM with model: {OLLAMA_MODEL}")
         return ollama_llm
     except Exception as e:

query_handler.py

@@ -3,22 +3,12 @@
 import os
 import re
 from dotenv import load_dotenv
-from mock_database import search_flights  # Import the correct function
+from mock_database import search_flights, check_ollama_availability
 
 # Load environment variables
 load_dotenv()
 OLLAMA_MODEL = os.getenv("OLLAMA_MODEL", "llama2:latest")  # Load model dynamically
 
-def check_ollama_availability():
-    """Check if the Ollama server is available before calling it."""
-    import requests
-    try:
-        response = requests.get("http://localhost:11434/api/tags", timeout=3)
-        return response.status_code == 200
-    except requests.RequestException:
-        print("⚠️ Ollama server is not available.")
-        return False
-
 OLLAMA_AVAILABLE = check_ollama_availability()
 
 def extract_entities_llama2(query):

requirements.txt

@@ -0,0 +1,44 @@
+aiohttp
+altair
+anyio
+attrs
+blinker
+cachetools
+certifi
+charset-normalizer
+click
+colorama
+dataclasses-json
+exceptiongroup
+gitdb
+GitPython
+h11
+httpx
+idna
+Jinja2
+joblib
+jsonschema
+langchain
+langchain-community
+langchain-core
+langchain-ollama==0.2.3
+MarkupSafe
+marshmallow
+numpy
+ollama==0.4.7
+pandas
+pillow
+pydantic
+python-dateutil
+python-dotenv
+PyYAML
+requests
+scikit-learn
+scipy
+SQLAlchemy
+streamlit
+tenacity
+tqdm
+typing_extensions
+urllib3
+watchdog