Update app.py

app.py

@@ -114,30 +114,57 @@ def show_combined_repo_and_llm():
     summary = f"{extraction_status}\n\nStrengths:\n{strengths}\n\nWeaknesses:\n{weaknesses}"
     return combined_content, summary, df
 
+def go_to_analysis():
+    return gr.update(visible=False), gr.update(visible=True)
+
+def go_to_input():
+    return gr.update(visible=True), gr.update(visible=False)
+
 repo_id_input = gr.Textbox(label="Enter repo IDs (comma or newline separated)", lines=5, placeholder="repo1, repo2\nrepo3")
-df_output = gr.Dataframe(headers=["repo id", "strength", "weaknesses", "speciality", "relevance rating", "Usecase"])
+df_output = gr.Dataframe(headers=["repo id", "strength", "weaknesses", "speciality", "relevance rating", "Usecase"],
+    datatype=["str", "str", "str", "str", "str", "str"],
+    max_rows=10,
+    max_cols=6,
+    wrap=True
+)
 
 with gr.Blocks() as demo:
-    gr.Markdown("## Repo ID Input")
-    repo_id_box = repo_id_input.render()
-    df_box = df_output.render()
-    submit_btn = gr.Button("Submit Repo IDs")
-    submit_btn.click(process_repo_input_and_store, inputs=repo_id_box, outputs=df_box)
+    page_state = gr.State(0)
+
+    # --- Page 1: Input ---
+    with gr.Column(visible=True) as input_page:
+        gr.Markdown("## Enter Keyword or Repo IDs")
+        keyword_input = gr.Textbox(label="Enter keyword to search repos", lines=1, placeholder="e.g. audio")
+        keyword_btn = gr.Button("Search and Update Repo List")
+        repo_id_box = repo_id_input.render()
+        df_box = df_output.render()
+        submit_btn = gr.Button("Submit Repo IDs")
+        next_btn = gr.Button("Next: Go to Analysis")
 
-    gr.Markdown("---")
-    gr.Markdown("## Keyword Search for Repos")
-    keyword_input = gr.Textbox(label="Enter keyword to search repos", lines=1, placeholder="e.g. audio")
-    keyword_btn = gr.Button("Search and Update Repo List")
+    # --- Page 2: Analysis ---
+    with gr.Column(visible=False) as analysis_page:
+        gr.Markdown("## Combine and Display Repo Files")
+        combine_btn = gr.Button("Download, Combine & Show .py/.md Files from Next Repo and Analyze")
+        combined_txt = gr.Textbox(label="Combined Repo Files", lines=20)
+        llm_output_txt = gr.Textbox(label="LLM Analysis Output", lines=10)
+        df_display = gr.Dataframe(
+            headers=["repo id", "strength", "weaknesses", "speciality", "relevance rating", "Usecase"],
+            datatype=["str", "str", "str", "str", "str", "str"],
+            max_rows=10,
+            max_cols=6,
+            wrap=True
+        )
+        back_btn = gr.Button("Back to Input")
+
+    # Button logic to switch pages
+    next_btn.click(go_to_analysis, inputs=None, outputs=[input_page, analysis_page])
+    back_btn.click(go_to_input, inputs=None, outputs=[input_page, analysis_page])
+
+    # Keyword and repo input logic
     keyword_btn.click(keyword_search_and_update, inputs=keyword_input, outputs=df_box)
+    submit_btn.click(process_repo_input_and_store, inputs=repo_id_box, outputs=df_box)
 
-    gr.Markdown("---")
-    gr.Markdown("## Combine and Display Repo Files")
-    combine_btn = gr.Button("Download, Combine & Show .py/.md Files from Next Repo and Analyze")
-    combined_txt = gr.Textbox(label="Combined Repo Files", lines=20)
-    llm_output_txt = gr.Textbox(label="LLM Analysis Output", lines=10)
-    df_display = gr.Dataframe(
-        headers=["repo id", "strength", "weaknesses", "speciality", "relevance rating", "Usecase"]
-    )
+    # Analysis logic
     combine_btn.click(show_combined_repo_and_llm, inputs=None, outputs=[combined_txt, llm_output_txt, df_display])
 
 demo.launch()