Spaces:

zamal
/

DeepGit-lite

Running on Zero

App Files Files Community

zamalali commited on Mar 31

Commit

5ab8b50

1 Parent(s): ae13177

Refactor DeepGit Lite UI to enhance layout, improve user interaction, and provide immediate feedback during processing.

Browse files

Files changed (1) hide show

app.py +47 -57

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ import time
 import threading
 import logging
 import spaces
-from main import run_repository_ranking  # Import the repository ranking function
 # ---------------------------
 # Global Logging Buffer Setup
@@ -37,36 +37,8 @@ def filter_logs(logs):
             last_was_fetching = False
     return filtered
-# ---------------------------
-# Minimal Title, Favicon & Description
-# ---------------------------
-favicon_html = """
-<head>
-<link rel="icon" type="image/x-icon" href="file/assets/deepgit.ico">
-<title>DeepGit Lite Research Agent</title>
-</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>
-"""
-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;">
@@ -136,32 +108,44 @@ def stream_lite_workflow(topic):
     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():
-        research_input = gr.Textbox(
-            label="Research Topic",
-            placeholder="Enter your research topic here, e.g., 'Fine tuning Instruction tuned LLama models...'",
-            lines=3
-        )
-        run_button = gr.Button("Run DeepGit Lite", variant="primary")
-        status_display = gr.Markdown("")
-        detail_display = gr.HTML("")
-        output_html = gr.HTML("")
-        state = gr.State([])
-    # 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):
-            yield status, details
     run_button.click(
         fn=lite_runner,
         inputs=[research_input],
@@ -169,7 +153,7 @@ with gr.Blocks() as demo:
         api_name="deepgit_lite",
         show_progress=True
     )
     research_input.submit(
         fn=lite_runner,
         inputs=[research_input],
@@ -177,7 +161,13 @@ with gr.Blocks() as demo:
         api_name="deepgit_lite_submit",
         show_progress=True
     )
-    gr.HTML(footer)
 demo.queue(max_size=10).launch()

 import threading
 import logging
 import spaces
+from main import run_repository_ranking  # Your repository ranking function
 # ---------------------------
 # Global Logging Buffer Setup
             last_was_fetching = False
     return filtered
 def parse_result_to_html(raw_result: str) -> str:
+    # Parses raw string output to an HTML table
     entries = raw_result.strip().split("Final Rank:")
     html = """
     <table border="1" style="width:100%; border-collapse: collapse;">
     html_result = parse_result_to_html(raw_result)
     yield "", html_result
+def lite_runner(topic):
+    # Provide immediate feedback upon button press.
+    yield "Workflow started", "<p>Processing your request. Please wait...</p>"
+    for status, details in stream_lite_workflow(topic):
+        yield status, details
 # ---------------------------
+# App UI Setup Using Gradio Soft Theme
 # ---------------------------
+with gr.Blocks(theme=gr.themes.Soft(), title="DeepGit Lite", fill_width=True) as demo:
+    gr.HTML(
+        """
+        <head>
+            <link rel="icon" type="image/x-icon" href="file/assets/deepgit.ico">
+            <title>DeepGit Lite Research Agent</title>
+        </head>
+        """
+    )
+    gr.Markdown(
+        """
+        # DeepGit Lite
+        Explore GitHub repositories with deep semantic search.
+        Check out our [GitHub](https://github.com/zamalali/DeepGit) for more details.
+        """
+    )
+    with gr.Row():
+        with gr.Column(scale=2):
+            research_input = gr.Textbox(
+                label="Research Query",
+                placeholder="Enter your research topic here, e.g., 'data augmentation pipelines for LLM fine-tuning'",
+                lines=3
+            )
+            run_button = gr.Button("Run DeepGit Lite", variant="primary")
+        with gr.Column(scale=3):
+            status_display = gr.Markdown(label="Status")
+            detail_display = gr.HTML(label="Results")
     run_button.click(
         fn=lite_runner,
         inputs=[research_input],
         api_name="deepgit_lite",
         show_progress=True
     )
     research_input.submit(
         fn=lite_runner,
         inputs=[research_input],
         api_name="deepgit_lite_submit",
         show_progress=True
     )
+    gr.HTML(
+        """
+        <div>
+            Made with ❤️ by <b>Zamal</b>
+        </div>
+        """
+    )
 demo.queue(max_size=10).launch()