Spaces:

zamal
/

DeepGit-lite

Running on Zero

App Files Files Community

zamalali commited on Mar 31

Commit

2da0a7b

1 Parent(s): 777083e

Enhance DeepGit Lite UI and functionality by integrating GPU support for repository ranking, refining HTML structure, and improving user feedback mechanisms.

Browse files

Files changed (1) hide show

app.py +27 -57

app.py CHANGED Viewed

@@ -2,7 +2,8 @@ import gradio as gr
 import time
 import threading
 import logging
-from main import run_repository_ranking  # Import the new function from main.py
 # ---------------------------
 # Global Logging Buffer Setup
@@ -37,7 +38,7 @@ def filter_logs(logs):
     return filtered
 # ---------------------------
-# Title, Favicon & Description
 # ---------------------------
 favicon_html = """
 <head>
@@ -47,23 +48,20 @@ favicon_html = """
 """
 title = """
-<div style="text-align: center; margin-top: 20px;">
-  <h1 style="font-size: 36px; display: inline-flex; align-items: center; gap: 16px;">
-    <img src="https://img.icons8.com/?size=100&id=118557&format=png&color=000000" width="64" />
-    <span>DeepGit Lite</span>
-  </h1>
-  <p style="font-size: 18px; color: #555; margin-top: 10px;">
-    ⚙️ A lightweight GitHub research agent for deep semantic search and ranking.
-  </p>
 </div>
 """
-description = """<p align="center">
 DeepGit Lite is a streamlined tool for semantic search on GitHub repositories. It retrieves repositories using dense retrieval, ranks them by similarity, and then presents the top results.
-</p>"""
 consent_text = """
-<div style="padding: 10px; text-align: center;">
   <p>
     By using DeepGit Lite, you consent to temporary processing of your query for semantic search and ranking purposes.
   </p>
@@ -75,42 +73,15 @@ consent_text = """
 """
 footer = """
-<div style="text-align: center; margin-top: 40px; font-size: 13px; color: #888;">
-    Made with <span style="color: crimson;">❤️</span> by <b>Zamal</b>
 </div>
 """
-# ---------------------------
-# HTML Table Renderer for Results
-# ---------------------------
-def format_percent(value):
-    try:
-        return f"{float(value) * 100:.1f}%"
-    except:
-        return value
 def parse_result_to_html(raw_result: str) -> str:
     entries = raw_result.strip().split("Final Rank:")
     html = """
-    <style>
-        table {
-            width: 100%;
-            border-collapse: collapse;
-            margin: 1em 0;
-            font-size: 14px;
-        }
-        th, td {
-            padding: 12px 15px;
-            border: 1px solid #ddd;
-            text-align: left;
-            vertical-align: top;
-        }
-        th {
-            background-color: #f4f4f4;
-        }
-        tr:hover { background-color: #f9f9f9; }
-    </style>
-    <table>
         <thead>
             <tr>
                 <th>Rank</th>
@@ -141,10 +112,14 @@ def parse_result_to_html(raw_result: str) -> str:
     return html
 # ---------------------------
-# Background Workflow Runner
 # ---------------------------
 def run_lite_workflow(topic, result_container):
-    result = run_repository_ranking(topic)
     result_container["raw_result"] = result
 def stream_lite_workflow(topic):
@@ -169,30 +144,23 @@ def stream_lite_workflow(topic):
     workflow_thread.join()
     with LOG_BUFFER_LOCK:
         final_logs = LOG_BUFFER[:]
-    filtered_final = filter_logs(final_logs)
     raw_result = result_container.get("raw_result", "No results returned.")
     html_result = parse_result_to_html(raw_result)
     yield "", html_result
 # ---------------------------
-# App UI Setup for DeepGit Lite
 # ---------------------------
-with gr.Blocks(
-    css="""
-        #main_container { margin: auto; max-width: 900px; }
-        footer, footer * { display: none !important; }
-    """
-) as demo:
     gr.HTML(favicon_html)
     gr.HTML(title)
     gr.HTML(description)
-    with gr.Column(elem_id="user_consent_container") as consent_block:
         gr.HTML(consent_text)
         agree_button = gr.Button("I Agree", variant="primary")
-    with gr.Column(elem_id="main_container", visible=False) as main_block:
         research_input = gr.Textbox(
             label="Research Topic",
             placeholder="Enter your research topic here, e.g., 'Fine tuning Instruction tuned LLama models...'",
@@ -209,7 +177,7 @@ with gr.Blocks(
     agree_button.click(fn=enable_main, inputs=[], outputs=[consent_block, main_block], queue=False)
-    # Added initial yield for immediate feedback when the button is pressed.
     def lite_runner(topic):
         yield "Workflow started", "<p>Processing your request. Please wait...</p>"
         for status, details in stream_lite_workflow(topic):
@@ -232,4 +200,6 @@ with gr.Blocks(
     )
     gr.HTML(footer)
 demo.queue(max_size=10).launch()

 import time
 import threading
 import logging
+import spaces
+from main import run_repository_ranking  # Import the repository ranking function
 # ---------------------------
 # Global Logging Buffer Setup
     return filtered
 # ---------------------------
+# Minimal Title, Favicon & Description
 # ---------------------------
 favicon_html = """
 <head>
 """
 title = """
+<div>
+  <h1>DeepGit Lite</h1>
+  <p>⚙️ A lightweight GitHub research agent for deep semantic search and ranking.</p>
 </div>
 """
+description = """
+<p>
 DeepGit Lite is a streamlined tool for semantic search on GitHub repositories. It retrieves repositories using dense retrieval, ranks them by similarity, and then presents the top results.
+</p>
+"""
 consent_text = """
+<div>
   <p>
     By using DeepGit Lite, you consent to temporary processing of your query for semantic search and ranking purposes.
   </p>
 """
 footer = """
+<div>
+    Made with ❤️ by <b>Zamal</b>
 </div>
 """
 def parse_result_to_html(raw_result: str) -> str:
     entries = raw_result.strip().split("Final Rank:")
     html = """
+    <table border="1" style="width:100%; border-collapse: collapse;">
         <thead>
             <tr>
                 <th>Rank</th>
     return html
 # ---------------------------
+# GPU-enabled Wrapper for Repository Ranking
 # ---------------------------
+@spaces.GPU(duration=180)
+def gpu_run_repo(topic: str):
+    return run_repository_ranking(topic)
 def run_lite_workflow(topic, result_container):
+    result = gpu_run_repo(topic)
     result_container["raw_result"] = result
 def stream_lite_workflow(topic):
     workflow_thread.join()
     with LOG_BUFFER_LOCK:
         final_logs = LOG_BUFFER[:]
     raw_result = result_container.get("raw_result", "No results returned.")
     html_result = parse_result_to_html(raw_result)
     yield "", html_result
 # ---------------------------
+# App UI Setup Using Default Gradio Theme
 # ---------------------------
+with gr.Blocks() as demo:
     gr.HTML(favicon_html)
     gr.HTML(title)
     gr.HTML(description)
+    with gr.Column() as consent_block:
         gr.HTML(consent_text)
         agree_button = gr.Button("I Agree", variant="primary")
+    with gr.Column(visible=False) as main_block:
         research_input = gr.Textbox(
             label="Research Topic",
             placeholder="Enter your research topic here, e.g., 'Fine tuning Instruction tuned LLama models...'",
     agree_button.click(fn=enable_main, inputs=[], outputs=[consent_block, main_block], queue=False)
+    # Yield an initial status message for immediate feedback
     def lite_runner(topic):
         yield "Workflow started", "<p>Processing your request. Please wait...</p>"
         for status, details in stream_lite_workflow(topic):
     )
     gr.HTML(footer)
 demo.queue(max_size=10).launch()