Update app.py

app.py

@@ -191,51 +191,51 @@ def get_gemini_flash_response(prompt: str) -> str:
 
 
 
-############################################
-# Gradio Interface Functions
-############################################
-
-# For file content retrieval, we now use the file path directly.
-def get_file_content_for_choice(github_url: str, file_path: str):
-    try:
-        owner, repo = extract_repo_info(github_url)
-    except Exception as e:
-        return str(e)
-    content = get_file_content(owner, repo, file_path)
-    return content, file_path
-
-def chat_with_file(github_url: str, file_path: str, user_query: str):
-    # Retrieve file content using the file path directly.
-    result = get_file_content_for_choice(github_url, file_path)
-    if isinstance(result, str):
-        return result  # Return error message if occurred.
-    file_content, selected_file = result
-    
-    # Preprocess file content and extract context.
-    preprocessed = preprocess_text(file_content)
-    context_snippet = preprocessed[:5000]  # Use first 1000 characters as context.
-    
-    # Generate the prompt based on context and user query.
-    prompt = generate_prompt(user_query, [context_snippet])
-    
-    # Use Gemini Flash to generate a response.
-    llm_response = get_gemini_flash_response(prompt)
-    
-    return f"File: {selected_file}\n\nLLM Response:\n{llm_response}"
-
-
-def load_repo_contents_backend(github_url: str):
-    try:
-        owner, repo = extract_repo_info(github_url)
-    except Exception as e:
-        return f"Error: {str(e)}"
-    repo_data = get_repo_metadata(owner, repo)
-    default_branch = repo_data.get("default_branch", "main")
-    tree_data = get_repo_tree(owner, repo, default_branch)
-    if "tree" not in tree_data:
-        return "Error: Could not fetch repository tree."
-    file_list = [item["path"] for item in tree_data["tree"] if item["type"] == "blob"]
-    return file_list
+# ############################################
+# # Gradio Interface Functions
+# ############################################
+
+# # For file content retrieval, we now use the file path directly.
+# def get_file_content_for_choice(github_url: str, file_path: str):
+#     try:
+#         owner, repo = extract_repo_info(github_url)
+#     except Exception as e:
+#         return str(e)
+#     content = get_file_content(owner, repo, file_path)
+#     return content, file_path
+
+# def chat_with_file(github_url: str, file_path: str, user_query: str):
+#     # Retrieve file content using the file path directly.
+#     result = get_file_content_for_choice(github_url, file_path)
+#     if isinstance(result, str):
+#         return result  # Return error message if occurred.
+#     file_content, selected_file = result
+    
+#     # Preprocess file content and extract context.
+#     preprocessed = preprocess_text(file_content)
+#     context_snippet = preprocessed[:5000]  # Use first 1000 characters as context.
+    
+#     # Generate the prompt based on context and user query.
+#     prompt = generate_prompt(user_query, [context_snippet])
+    
+#     # Use Gemini Flash to generate a response.
+#     llm_response = get_gemini_flash_response(prompt)
+    
+#     return f"File: {selected_file}\n\nLLM Response:\n{llm_response}"
+
+
+# def load_repo_contents_backend(github_url: str):
+#     try:
+#         owner, repo = extract_repo_info(github_url)
+#     except Exception as e:
+#         return f"Error: {str(e)}"
+#     repo_data = get_repo_metadata(owner, repo)
+#     default_branch = repo_data.get("default_branch", "main")
+#     tree_data = get_repo_tree(owner, repo, default_branch)
+#     if "tree" not in tree_data:
+#         return "Error: Could not fetch repository tree."
+#     file_list = [item["path"] for item in tree_data["tree"] if item["type"] == "blob"]
+#     return file_list
 
 ############################################
 # Gradio Interface Setup
@@ -291,7 +291,6 @@ def load_repo_contents_backend(github_url: str):
 # demo.launch(share=True)
 
 
-# Retrieve file content using file path.
 def get_file_content_for_choice(github_url: str, file_path: str):
     try:
         owner, repo = extract_repo_info(github_url)
@@ -306,8 +305,7 @@ def chat_with_file(github_url: str, file_path: str, user_query: str):
         return result  # Return error message if occurred.
     file_content, selected_file = result
     preprocessed = preprocess_text(file_content)
-    # Use the first 1000 characters as context (adjust as needed).
-    context_snippet = preprocessed[:1000]
+    context_snippet = preprocessed[:1000]  # Use the first 1000 characters as context
     prompt = generate_prompt(user_query, [context_snippet])
     llm_response = get_gemini_flash_response(prompt)
     return f"File: {selected_file}\n\nLLM Response:\n{llm_response}"
@@ -320,7 +318,7 @@ def load_repo_contents_backend(github_url: str):
     repo_data = get_repo_metadata(owner, repo)
     default_branch = repo_data.get("default_branch", "main")
     tree_data = get_repo_tree(owner, repo, default_branch)
-    if "tree" not in tree_data:
+    if not isinstance(tree_data, dict) or "tree" not in tree_data:
         return "Error: Could not fetch repository tree."
     file_list = [item["path"] for item in tree_data["tree"] if item["type"] == "blob"]
     return file_list
@@ -331,44 +329,36 @@ def load_repo_contents_backend(github_url: str):
 
 with gr.Blocks() as demo:
     gr.Markdown("# RepoChat - Chat with Repository Files")
-    
+
     with gr.Row():
         with gr.Column(scale=1):
             gr.Markdown("### Repository Information")
             github_url_input = gr.Textbox(label="GitHub Repository URL", placeholder="https://github.com/username/repository")
             load_repo_btn = gr.Button("Load Repository Contents")
-            # File dropdown without preselection.
-            file_dropdown = gr.Dropdown(label="Select a File", interactive=True, value="", choices=[])
-            # Display file content in a textbox.
-            repo_content_output = gr.Textbox(label="File Content", interactive=False, lines=30)
+            file_dropdown = gr.Dropdown(label="Select a File", interactive=True, choices=[])
+            repo_content_output = gr.Textbox(label="File Content", interactive=False, lines=20)
+        
         with gr.Column(scale=2):
             gr.Markdown("### Chat Interface")
-            # Use Chatbot component for conversation history.
             chat_output = gr.Chatbot(label="Chat Conversation")
             chat_query_input = gr.Textbox(label="Your Query", placeholder="Type your query here")
             chat_btn = gr.Button("Send Query")
-    
-    # Callback: Update file dropdown choices.
+
+    # State to hold conversation history
+    conversation_history = gr.State([])
+
     def update_file_dropdown(github_url):
         files = load_repo_contents_backend(github_url)
-        if isinstance(files, str):  # If an error message was returned.
-            print("Error loading files:", files)
-            return {"choices": [], "value": ""}
-        print("Files loaded:", files)
-        return {"choices": files, "value": ""}
-    
-    load_repo_btn.click(fn=update_file_dropdown, inputs=[github_url_input], outputs=[file_dropdown])
-    
-    # Callback: Update repository content display when a file is selected.
+        if isinstance(files, str):  # Error message
+            return gr.update(choices=[]), files
+        return gr.update(choices=files), ""
+
     def update_repo_content(github_url, file_choice):
         if not file_choice:
             return "No file selected."
         content, _ = get_file_content_for_choice(github_url, file_choice)
         return content
-    
-    file_dropdown.change(fn=update_repo_content, inputs=[github_url_input, file_dropdown], outputs=[repo_content_output])
-    
-    # Callback: Process chat query and update conversation history.
+
     def process_chat(github_url, file_choice, chat_query, history):
         if not file_choice:
             history.append(("System", "Please select a file first."))
@@ -376,13 +366,9 @@ with gr.Blocks() as demo:
         response = chat_with_file(github_url, file_choice, chat_query)
         history.append((chat_query, response))
         return history, history
-    
-    conversation_history = gr.State([])
-    
-    chat_btn.click(
-        fn=process_chat, 
-        inputs=[github_url_input, file_dropdown, chat_query_input, conversation_history], 
-        outputs=[chat_output, conversation_history]
-    )
-    
-demo.launch(share=True, server_port=7862)
+
+    load_repo_btn.click(fn=update_file_dropdown, inputs=[github_url_input], outputs=[file_dropdown, repo_content_output])
+    file_dropdown.change(fn=update_repo_content, inputs=[github_url_input, file_dropdown], outputs=[repo_content_output])
+    chat_btn.click(fn=process_chat, inputs=[github_url_input, file_dropdown, chat_query_input, conversation_history], outputs=[chat_output, conversation_history])
+
+demo.launch(server_port=7862)