Update app.py

app.py

@@ -1,34 +1,113 @@
-from docs import generate_docs
 import gradio as gr
+import requests
+from urllib.parse import urlparse
 
-DESCRIPTION = "GitHub Repo Documentation Generator"
+try:
+    from groq import Groq
+except ImportError:
+    print("Error: The Groq Python client is required. Install with 'pip install groq'")
 
-# Define the Gradio Blocks interface.
-with gr.Blocks(theme=gr.themes.Origin(primary_hue="red", secondary_hue="pink"), title="GitHub Repo Documentation Generator") as demo:
-    gr.HTML(
-        f"""<h1><span>{DESCRIPTION}</span></h1>""",
-        elem_id="title",
-    )
-    gr.Markdown(
-        "Enter a GitHub repository URL or ID (in the format `user/repo`) below. "
-        "This tool fetches repository details and uses the Groq API to generate documentation."
-    )
-    gr.Markdown("liek this spaces if this spaces helpfull")
-    with gr.Row():
-        repo_id_input = gr.Textbox(
-            label="GitHub Repository URL or ID",
+def extract_repo_id(repo_input):
+    if repo_input.startswith(("http://", "https://")):
+        parsed = urlparse(repo_input)
+        path = parsed.path.strip("/")
+        parts = path.split("/")
+        if len(parts) >= 2:
+            return f"{parts[0]}/{parts[1]}"
+        return None
+    else:
+        if "/" in repo_input:
+            return repo_input
+        return None
+
+def get_repo_info(repo_id):
+    try:
+        # Try model endpoint
+        model_response = requests.get(f"https://huggingface.co/api/models/{repo_id}")
+        if model_response.status_code == 200:
+            return model_response.json(), "model"
+        
+        # Try dataset endpoint
+        dataset_response = requests.get(f"https://huggingface.co/api/datasets/{repo_id}")
+        if dataset_response.status_code == 200:
+            return dataset_response.json(), "dataset"
+        
+        return None, None
+    except Exception as e:
+        print(f"Error fetching repo info: {e}")
+        return None, None
+
+def generate_readme(repo_input, api_key):
+    try:
+        # Validate and extract repo ID
+        repo_id = extract_repo_id(repo_input)
+        if not repo_id:
+            return "Invalid repository format. Please use 'user/repo' or a Hugging Face URL"
+        
+        # Get repository information
+        repo_info, repo_type = get_repo_info(repo_id)
+        if not repo_info:
+            return "Repository not found or inaccessible"
+        
+        # Prepare prompt for Groq
+        prompt = f"""Generate a professional README.md for the Hugging Face {repo_type} repository {repo_id}.
+        Use the following information:
+        - Author: {repo_info.get('author', 'Unknown')}
+        - Description: {repo_info.get('description', 'No description')}
+        - Tags: {', '.join(repo_info.get('tags', []))}
+        - License: {repo_info.get('license', 'Unknown')}
+        
+        Include these sections:
+        1. Overview
+        2. Installation
+        3. Usage
+        4. Examples
+        5. License
+        6. Citation (if applicable)
+        
+        Format the README in proper Markdown with code blocks where appropriate."""
+        
+        # Initialize Groq client
+        client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
+        
+        # Generate completion
+        completion = client.chat.completions.create(
+            messages=[{"role": "user", "content": prompt}],
+            model="mixtral-8x7b-32768",
+            temperature=0.3,
+            max_tokens=1024
         )
-        generate_button = gr.Button("Generate Documentation")
+        
+        return completion.choices[0].message.content
+        
+    except Exception as e:
+        return f"Error generating README: {str(e)}"
 
-    output_box = gr.Textbox(
-        label="Generated Documentation",
-        lines=20, 
-        interactive=True,
+with gr.Blocks(title="HF Repo README Generator") as demo:
+    gr.Markdown("# 🚀 Hugging Face Repository README Generator")
+    
+    with gr.Row():
+        with gr.Column():
+            repo_input = gr.Textbox(
+                label="Hugging Face Repository URL or ID",
+                placeholder="Enter 'user/repo' or full URL...",
+                max_lines=1
+            )
+            api_key = gr.Textbox(
+                label="Groq API Key",
+                type="password",
+                placeholder="Enter your Groq API key..."
+            )
+            submit_btn = gr.Button("Generate README", variant="primary")
+    
+    with gr.Row():
+        output = gr.Markdown(label="Generated README")
+    
+    submit_btn.click(
+        fn=generate_readme,
+        inputs=[repo_input, api_key],
+        outputs=output
     )
-    gr.Markdown("<div style='text-align: center;'>Made with ❤ by <a href='https://huggingface.co/theNeofr'>NeoPy</a></div>")
-
-    # When the button is clicked, call the generate_docs function.
-    generate_button.click(fn=generate_docs, inputs=repo_id_input, outputs=output_box)
 
-# Launch the Gradio app.
-demo.launch(debug=True)
+if __name__ == "__main__":
+    demo.launch()