Update app.py

app.py

@@ -7,14 +7,12 @@ import json
 import google.generativeai as genai
 
 # --- IMPORTANT: Configure your API Key ---
-# For local testing, you can set this directly.
-# For Hugging Face Spaces, you will set this in "Secrets".
-# os.environ['GOOGLE_API_KEY'] = "YOUR_API_KEY_HERE" 
 try:
     genai.configure(api_key=os.environ.get("GOOGLE_API_KEY"))
 except AttributeError:
     print("WARNING: GOOGLE_API_KEY secret not set. LLM functionality will fail.")
 
+# Create the Flask app instance
 app = Flask(__name__)
 
 # The HTML template remains the same
@@ -81,13 +79,12 @@ def generate_readme_with_llm(repo_path):
     """
     Analyzes the repo and uses an LLM to generate the README content.
     """
-    # 1. Gather context: file structure and content of key files
     file_structure = ""
     file_contents = ""
-    file_limit = 5  # Limit number of files to read to save tokens
-    
+    file_limit = 5
+    char_limit_per_file = 2000
+
     for root, _, files in os.walk(repo_path):
-        # Ignore git directory
         if '.git' in root:
             continue
         
@@ -98,51 +95,37 @@ def generate_readme_with_llm(repo_path):
         sub_indent = ' ' * 4 * (level + 1)
         for f in files[:file_limit]:
             file_structure += f"{sub_indent}{f}\n"
-            # Read content of a few key files
             try:
                 with open(os.path.join(root, f), 'r', errors='ignore') as file:
-                    content = file.read(2000) # Read first 2000 chars
+                    content = file.read(char_limit_per_file)
                     file_contents += f"\n--- Start of {f} ---\n{content}\n--- End of {f} ---\n"
             except Exception:
-                continue # Skip files we can't read
+                continue
 
-    # 2. Create a detailed prompt for the LLM
     prompt = f"""
     You are an expert technical writer tasked with creating a high-quality README.md for a GitHub repository.
     Analyze the following repository context and generate a comprehensive and user-friendly README.
 
     **Repository Context:**
-
     **1. File Structure:**
     ```
     {file_structure}
     ```
-
     **2. Content of Key Files:**
     ```
     {file_contents}
     ```
-
     **Instructions:**
-    Based on the context provided, please generate a README.md file with the following sections:
-    1.  **Project Title:** An appropriate title for the project.
-    2.  **About the Project:** A concise but informative paragraph describing the project's purpose and the technology it uses. Infer this from the file contents.
-    3.  **Getting Started:** Clear, step-by-step instructions for installation and setup. Infer the package manager (pip, npm, etc.) and run commands.
-    4.  **Usage:** A simple example of how to run the application.
-
-    **Rules:**
-    - The output must be in valid Markdown format.
-    - Be professional and clear.
-    - If you cannot determine a piece of information (like a specific run command), suggest a common default and state that it might need to be verified.
-    - Do not invent features that are not evident from the provided context.
+    Generate a README.md with these sections: Project Title, About the Project, Getting Started, and Usage.
+    - Infer the project purpose, technologies, and setup commands from the files.
+    - The output must be valid Markdown.
+    - If a command is unknown, suggest a common default (e.g., `npm install`).
     """
 
-    # 3. Call the LLM
     print("Sending request to Gemini API...")
     model = genai.GenerativeModel('gemini-1.5-flash-latest')
     response = model.generate_content(prompt)
     
-    # Clean up the response - sometimes the model wraps it in ```markdown
     readme_text = response.text.strip()
     if readme_text.startswith("```markdown"):
         readme_text = readme_text[10:]
@@ -157,31 +140,42 @@ def index():
 
 @app.route('/generate', methods=['POST'])
 def generate():
-    repo_url = request.get_json().get('url')
-    if not repo_url or not repo_url.startswith("[https://github.com/](https://github.com/)"):
-        return jsonify({"error": "A valid GitHub repository URL is required."}), 400
+    data = request.get_json()
+    if not data or 'url' not in data:
+        return jsonify({"error": "Request body must be JSON with a 'url' key."}), 400
+        
+    repo_url = data.get('url')
+    
+    # --- BUG FIX: Make URL validation more flexible ---
+    if not repo_url or "github.com/" not in repo_url.lower():
+        print(f"Validation failed for URL: {repo_url}")
+        return jsonify({"error": "A valid public GitHub repository URL is required."}), 400
 
     if not os.environ.get("GOOGLE_API_KEY"):
          return jsonify({"error": "Server is missing the GOOGLE_API_KEY. Cannot contact the LLM."}), 500
 
     temp_dir = tempfile.mkdtemp()
     try:
+        print(f"Cloning repository: {repo_url} into {temp_dir}")
         git.Repo.clone_from(repo_url, temp_dir)
+        print("Cloning successful.")
         
-        cloned_folders = [d for d in os.listdir(temp_dir) if os.path.isdir(os.path.join(temp_dir, d))]
-        if not cloned_folders:
-            raise Exception("Cloning seems to have failed, the directory is empty.")
-        
-        repo_root_path = os.path.join(temp_dir, cloned_folders[0])
-        
-        # --- NEW: Call the LLM-powered function ---
-        readme_content = generate_readme_with_llm(repo_root_path)
+        readme_content = generate_readme_with_llm(temp_dir)
         
         return jsonify({"readme": readme_content})
 
+    except git.exc.GitCommandError as e:
+        error_message = str(e).lower()
+        print(f"Git error: {error_message}")
+        if "authentication failed" in error_message or "not found" in error_message:
+             return jsonify({"error": "Failed to clone. Please ensure the URL is correct and the repository is public."}), 400
+        else:
+             return jsonify({"error": f"A Git error occurred during cloning."}), 500
     except Exception as e:
-        return jsonify({"error": f"An error occurred: {str(e)}"}), 500
+        print(f"An unexpected error occurred: {e}")
+        return jsonify({"error": f"An unexpected error occurred on the server."}), 500
     finally:
+        print(f"Cleaning up temporary directory: {temp_dir}")
         shutil.rmtree(temp_dir)
 
 if __name__ == '__main__':