Add Dockerfile, update agents and app for environment variable support, and enhance requirements

Dockerfile

@@ -0,0 +1,32 @@
+FROM python:3.11-slim
+
+WORKDIR /app
+
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    build-essential \
+    curl \
+    software-properties-common \
+    git \
+    && rm -rf /var/lib/apt/lists/*
+
+# Set environment variables
+ENV PYTHONDONTWRITEBYTECODE=1
+ENV PYTHONUNBUFFERED=1
+ENV PORT=7860
+
+# Copy requirements and install Python dependencies
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Copy the rest of the application
+COPY . .
+
+# Set default API key (this will be overridden by Hugging Face secrets)
+ENV TOGETHER_API_KEY=""
+
+# Expose the port the app runs on
+EXPOSE 7860
+
+# Command to run the application
+CMD ["python", "app.py"]

agents.py

@@ -5,21 +5,42 @@ import json
 warnings.filterwarnings("ignore")
 from together import Together
 
-# Load configuration from config.json
+# Load configuration from config.json or environment variables
 def load_config():
-    config_path = os.path.join(os.path.dirname(__file__), "config.json")
-    with open(config_path, "r") as f:
-        return json.load(f)
+    # First try to get API key from environment variable (for Hugging Face deployment)
+    together_api_key = os.environ.get("TOGETHER_API_KEY", "")
+    
+    # If not found in environment, try to load from config.json (for local development)
+    if not together_api_key:
+        try:
+            config_path = os.path.join(os.path.dirname(__file__), "config.json")
+            if os.path.exists(config_path):
+                with open(config_path, "r") as f:
+                    config = json.load(f)
+                    together_api_key = config.get("together_ai_token", "")
+                    model_name = config.get("model", "meta-llama/Llama-4-Maverick-17B-128E-Instruct-FP8")
+                    return {"together_ai_token": together_api_key, "model": model_name}
+        except Exception as e:
+            print(f"Error loading config.json: {str(e)}")
+    
+    # Return a config dictionary with the API key from environment variable
+    return {
+        "together_ai_token": together_api_key,
+        "model": "meta-llama/Llama-4-Maverick-17B-128E-Instruct-FP8"  # Default model
+    }
 
 # Get API key and model from config
 config = load_config()
 your_api_key = config["together_ai_token"]
 model = config.get("model", "meta-llama/Llama-4-Maverick-17B-128E-Instruct-FP8")  # Use default if not in config
-client = Together(api_key=your_api_key)
 
+# Initialize client only if we have an API key
+client = Together(api_key=your_api_key) if your_api_key else None
 
 def prompt_llm(prompt, show_cost=False):
     # This function allows us to prompt an LLM via the Together API
+    if not client:
+        return "Error: Together API client not initialized. Please check your API key."
 
     # Calculate the number of tokens
     tokens = len(prompt.split())
@@ -31,16 +52,19 @@ def prompt_llm(prompt, show_cost=False):
         print(f"Estimated cost for {model}: ${cost:.10f}\n")
 
     # Make the API call
-    response = client.chat.completions.create(
-        model=model,
-        messages=[{"role": "user", "content": prompt}],
-    )
-    return response.choices[0].message.content
+    try:
+        response = client.chat.completions.create(
+            model=model,
+            messages=[{"role": "user", "content": prompt}],
+        )
+        return response.choices[0].message.content
+    except Exception as e:
+        return f"Error calling Together API: {str(e)}"
 
 
 class SummarizerAgent:
     def __init__(self):
-        self.client = Together(api_key=your_api_key)
+        self.client = client
 
     def process(self, content):
         prompt = """SYSTEM: You are an expert code summarizer. 
@@ -64,7 +88,7 @@ class SummarizerAgent:
 
 class InsightAgent:
     def __init__(self):
-        self.client = Together(api_key=your_api_key)
+        self.client = client
 
     def process_text(self, summaries):
         # Process a list of summary texts directly
@@ -94,7 +118,7 @@ class InsightAgent:
 
 class RecommenderAgent:
     def __init__(self):
-        self.client = Together(api_key=your_api_key)
+        self.client = client
 
     def process(self, insights, summaries, user_goal, persona=""):
         prompt = """SYSTEM: You are an expert code consultant who provides actionable recommendations.
@@ -160,7 +184,7 @@ class RecommenderAgent:
 
 class QuestionGeneratorAgent:
     def __init__(self):
-        self.client = Together(api_key=your_api_key)
+        self.client = client
         
     def generate_questions(self, content, category, source):
         prompt_template_path = os.path.join(os.path.dirname(__file__), "PromptTemplate.json")
@@ -180,7 +204,7 @@ class QuestionGeneratorAgent:
 
 class CLISetupAgent:
     def __init__(self):
-        self.client = Together(api_key=your_api_key)
+        self.client = client
         
     def generate_setup_instructions(self, repo_content, repo_metadata):
         """Generate step-by-step CLI instructions to set up the environment for a repository."""
@@ -262,7 +286,7 @@ class ChatbotAgent:
     """Agent for answering questions about GitHub repositories."""
     
     def __init__(self):
-        self.client = Together(api_key=your_api_key)
+        self.client = client
     
     def answer_question(self, question, repo_content, repo_metadata, summaries=None, insights=None):
         """
@@ -400,7 +424,7 @@ class PRReviewAgent:
     """Agent for reviewing GitHub Pull Requests and providing professional code feedback."""
     
     def __init__(self):
-        self.client = Together(api_key=your_api_key)
+        self.client = client
     
     def review_pr(self, pr_details, target_branch_code):
         """

app.py

@@ -316,4 +316,7 @@ def verify_credentials():
         return jsonify({"valid": False, "error": "Invalid GitHub credentials"}), 401
 
 if __name__ == "__main__":
-    app.run(debug=True, port=5001, host="0.0.0.0")
+    # Use environment variables for port if available (needed for Hugging Face)
+    port = int(os.environ.get('PORT', 5001))
+    # Bind to 0.0.0.0 instead of 127.0.0.1 to be accessible from outside the container
+    app.run(debug=False, host='0.0.0.0', port=port)

requirements.txt

@@ -1,3 +1,8 @@
 flask==2.3.3
 requests==2.31.0
+beautifulsoup4==4.12.2
+together>=0.1.5
+bs4==0.0.1
+lxml>=5.3.0
+gunicorn==20.1.0
 duckduckgo-search==3.0.2

space.yml

@@ -0,0 +1,7 @@
+title: GitHub Helper
+emoji: 🛠️
+colorFrom: blue
+colorTo: purple
+sdk: docker
+pinned: false
+app_port: 7860