Update app.py

app.py

@@ -83,6 +83,41 @@ async def convert_to_text(url):
         return f"Error: {e}", "", None, [], [], [], [], []  # Return error message and empty data
 
 # Model to Text Converter
+async def fetch_model_info(model_url):
+    """Fetches model description and installation instructions."""
+    try:
+        if "huggingface.co" in model_url:
+            # Fetch model card from Hugging Face
+            response = await asyncio.to_thread(requests.get, model_url, timeout=5)
+            response.raise_for_status()
+            soup = BeautifulSoup(response.text, "html.parser")
+
+            # Extract model description
+            description = soup.find("div", {"class": "prose"}).get_text(strip=True) if soup.find("div", {"class": "prose"}) else "No description available."
+
+            # Generate installation instructions
+            model_name = model_url.split("/")[-1]
+            install_instructions = f"To install this model, run:\n```bash\npip install transformers\n```\nThen load the model in Python:\n```python\nfrom transformers import AutoModel, AutoTokenizer\nmodel = AutoModel.from_pretrained('{model_name}')\ntokenizer = AutoTokenizer.from_pretrained('{model_name}')\n```"
+
+            return description, install_instructions
+        elif "github.com" in model_url:
+            # Fetch README from GitHub
+            readme_url = f"{model_url}/raw/main/README.md"
+            response = await asyncio.to_thread(requests.get, readme_url, timeout=5)
+            response.raise_for_status()
+
+            # Extract description from README
+            description = response.text if response.text else "No description available."
+
+            # Generate installation instructions
+            install_instructions = f"To install this model, clone the repository:\n```bash\ngit clone {model_url}.git\ncd {model_url.split('/')[-1]}\n```"
+
+            return description, install_instructions
+        else:
+            return "Unsupported repository.", ""
+    except Exception as e:
+        return f"Error: {e}", ""
+
 async def fetch_model_file_content(model_url, file_path):
     """Fetches the content of a file from a model repository (Hugging Face or GitHub)."""
     try:
@@ -178,13 +213,22 @@ with gr.Blocks(css=css) as demo:
                 model_url_input = gr.Textbox(label="Model URL", placeholder="https://huggingface.co/... or https://github.com/...")
                 file_path_input = gr.Textbox(label="File Path", placeholder="e.g., config.json or README.md")
             
+            with gr.Row():
+                model_description_output = gr.Textbox(label="Model Description", interactive=False)
+                install_instructions_output = gr.Textbox(label="Installation Instructions", interactive=False)
+            
             with gr.Row():
                 model_content_output = gr.Textbox(label="File Content", interactive=True, elem_id="model-content-output")
             
             with gr.Row():
                 gr.HTML("<button onclick='copyCode(\"model-content-output\")'>Copy Code</button>")  # Add the "Copy Code" button
             
-            submit_model_button = gr.Button("Fetch File Content")
+            submit_model_button = gr.Button("Fetch Model Info and File Content")
+            submit_model_button.click(
+                fn=fetch_model_info,
+                inputs=[model_url_input],
+                outputs=[model_description_output, install_instructions_output]
+            )
             submit_model_button.click(
                 fn=fetch_model_file_content,
                 inputs=[model_url_input, file_path_input],