Upload 7 files

.gitignore

@@ -0,0 +1,71 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+
+# Virtual environments
+venv/
+env/
+ENV/
+.venv/
+.env/
+.virtualenv/
+
+# Environments
+.env
+.env.*
+!.env.example
+
+# IDE specific files
+.idea/
+.vscode/
+*.swp
+*.swo
+.DS_Store
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# Testing
+.coverage
+htmlcov/
+
+# Logs
+logs/
+*.log
+
+# Local development settings
+local_settings.py
+
+# Database files
+*.db
+*.sqlite3
+
+# Large binary files
+*.pkl
+*.h5
+*.npy
+*.npz
+
+venv/

.streamlit/config.toml

@@ -0,0 +1,9 @@
+[server]
+port = 8501
+address = "0.0.0.0"
+
+[browser]
+gatherUsageStats = false
+
+[global]
+developmentMode = false 

Dockerfile

@@ -9,13 +9,30 @@ RUN apt-get update && apt-get install -y \
     git \
     && rm -rf /var/lib/apt/lists/*
 
-COPY requirements.txt ./
-COPY src/ ./src/
+# Create a non-root user
+RUN useradd -m -u 1000 streamlit
 
-RUN pip3 install -r requirements.txt
+# Create necessary directories and set permissions
+RUN mkdir -p /app/.streamlit && \
+    chown -R streamlit:streamlit /app
+
+# Switch to non-root user
+USER streamlit
+
+COPY --chown=streamlit:streamlit requirements.txt ./
+COPY --chown=streamlit:streamlit src/ ./src/
+COPY --chown=streamlit:streamlit .streamlit/ ./.streamlit/
+
+RUN pip3 install --user -r requirements.txt
+
+# Set environment variables
+ENV STREAMLIT_SERVER_PORT=8501
+ENV STREAMLIT_SERVER_ADDRESS=0.0.0.0
+ENV STREAMLIT_BROWSER_GATHER_USAGE_STATS=false
+ENV STREAMLIT_GLOBAL_DEVELOPMENT_MODE=false
 
 EXPOSE 8501
 
 HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health
 
-ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]
+ENTRYPOINT ["python", "-m", "streamlit", "run", "src/streamlit_app.py"]

src/config.py

@@ -0,0 +1,31 @@
+"""Configuration for available Cerebras models."""
+
+from typing import Dict, TypedDict
+
+class ModelConfig(TypedDict):
+    name: str
+    tokens: int
+    developer: str
+
+MODELS: Dict[str, ModelConfig] = {
+    "llama3.1-8b": {
+        "name": "Llama3.1-8b",
+        "tokens": 8192,
+        "developer": "Meta"
+    },
+    "llama-3.3-70b": {
+        "name": "Llama-3.3-70b",
+        "tokens": 8192,
+        "developer": "Meta"
+    },
+    "llama-4-scout-17b-16e-instruct": {
+        "name": "Llama4 Scout",
+        "tokens": 8192,
+        "developer": "Meta"
+    },
+    "qwen-3-32b": {
+        "name": "Qwen 3 32B",
+        "tokens": 8192,
+        "developer": "Alibaba"
+    }
+} 

src/streamlit_app.py

@@ -1,28 +1,46 @@
 import streamlit as st
 import time
 import os
+import sys
+from config import MODELS
 
 # Attempt to import Cerebras SDK and specific error classes
+# Set up variables to capture debugging information
+import_error_details = ""
+sdk_import_paths = []
+
 try:
+    # Try to explicitly check if the cerebras module is available
+    import importlib.util
+    spec = importlib.util.find_spec("cerebras")
+    if spec is not None:
+        sdk_import_paths.append(f"cerebras module found at: {spec.origin}")
+    else:
+        sdk_import_paths.append("cerebras module not found in sys.path")
+    
+    # Capture Python's module search paths
+    sdk_import_paths.append("Python sys.path contains:")
+    for path in sys.path:
+        sdk_import_paths.append(f"  - {path}")
+    
+    # Now try the actual import
     from cerebras.cloud.sdk import Cerebras
-    from cerebras.cloud.sdk.errors import APIError, APIConnectionError, AuthenticationError
+    # Try to import error classes directly from the main SDK package
+    # The error classes are likely defined within the main SDK package
+    from cerebras.cloud.sdk import APIError, APIConnectionError, AuthenticationError
     CEREBRAS_SDK_AVAILABLE = True
-except ImportError:
+    sdk_import_paths.append("✅ Cerebras SDK import successful")
+except ImportError as e:
     CEREBRAS_SDK_AVAILABLE = False
+    import_error_details = str(e)
+    sdk_import_paths.append(f"❌ Import Error: {import_error_details}")
+    
     # Define dummy classes if SDK is not available, so the rest of the code doesn't break
     class Cerebras: pass
     class APIError(Exception): pass
     class APIConnectionError(APIError): pass
     class AuthenticationError(APIError): pass
 
-# --- Configuration ---
-MODELS = {
-    "llama3.1-8b": {"name": "Llama3.1-8b", "tokens": 8192, "developer": "Meta"},
-    "llama-3.3-70b": {"name": "Llama-3.3-70b", "tokens": 8192, "developer": "Meta"},
-    "llama-4-scout-17b-16e-instruct": {"name": "Llama4 Scout", "tokens": 8192, "developer": "Meta"},
-    "qwen-3-32b":{"name": "Qwen 3 32B", "tokens": 8192, "developer": "Qwen"},
-}
-
 # --- Helper Functions (Actual API Interaction) ---
 def get_cebras_response(api_key, model_id, current_prompt, chat_history_for_api):
     """
@@ -80,13 +98,22 @@ def get_cebras_response(api_key, model_id, current_prompt, chat_history_for_api)
 # --- Streamlit App ---
 st.set_page_config(page_title="Cerebras Chatbot", page_icon="🤖")
 
+st.write("Python Path:", sys.executable)
+st.write("Python Version:", sys.version)
+
+# Display detailed import debugging information
+st.expander("🔍 SDK Import Debug Information").write("\n".join(sdk_import_paths))
+
 st.title("🤖 Cerebras Powered Chatbot")
 
 if not CEREBRAS_SDK_AVAILABLE:
     st.error(
-        "The Cerebras SDK is not installed. Please install it by running `pip install cerebras-cloud-sdk` in your terminal and restart the app.",
+        f"The Cerebras SDK is not installed. Please install it by running `pip install cerebras-cloud-sdk` in your terminal and restart the app.\n\nError details: {import_error_details}",
         icon="🚨"
     )
+    
+    # Add a command to check the pip installation
+    st.code("python -m pip list | grep cerebras", language="bash")
     st.stop()
 
 st.caption("A Streamlit application for interacting with Cerebras models via `cerebras.cloud.sdk`.")
@@ -187,3 +214,4 @@ if prompt := st.chat_input("What would you like to ask?"):
 if st.sidebar.button("Clear Chat History"):
     st.session_state.messages = []
     st.rerun()
+