Update app.py

app.py

@@ -291,6 +291,44 @@ 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)
+    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):
+    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
+    preprocessed = preprocess_text(file_content)
+    # Use the first 1000 characters as context (adjust as needed).
+    context_snippet = preprocessed[:1000]
+    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}"
+
+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
+############################################
+
 with gr.Blocks() as demo:
     gr.Markdown("# RepoChat - Chat with Repository Files")
     
@@ -299,28 +337,29 @@ with gr.Blocks() as demo:
             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=[])
-            # For file content display, you can either use a Chatbot or a Textbox. Here we use Textbox.
+            # Display file content in a textbox.
             repo_content_output = gr.Textbox(label="File Content", interactive=False, lines=30)
         with gr.Column(scale=2):
             gr.Markdown("### Chat Interface")
-            # Use Chatbot component for conversation history
+            # 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 to update file dropdown choices.
+    # Callback: Update file dropdown choices.
     def update_file_dropdown(github_url):
         files = load_repo_contents_backend(github_url)
-        if isinstance(files, str):  # Error message
+        if isinstance(files, str):  # If an error message was returned.
             print("Error loading files:", files)
-            return gr.update(choices=[], value="")
+            return {"choices": [], "value": ""}
         print("Files loaded:", files)
-        return gr.update(choices=files, value="")  # No pre-selection.
+        return {"choices": files, "value": ""}
     
     load_repo_btn.click(fn=update_file_dropdown, inputs=[github_url_input], outputs=[file_dropdown])
     
-    # Callback to update repository content when a file is selected.
+    # Callback: Update repository content display when a file is selected.
     def update_repo_content(github_url, file_choice):
         if not file_choice:
             return "No file selected."
@@ -329,21 +368,21 @@ with gr.Blocks() as demo:
     
     file_dropdown.change(fn=update_repo_content, inputs=[github_url_input, file_dropdown], outputs=[repo_content_output])
     
-    # Callback to process the chat query and update conversation history.
+    # 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."))
             return history, history
         response = chat_with_file(github_url, file_choice, chat_query)
         history.append((chat_query, response))
-        return history, history  # Return updated conversation for both outputs.
+        return history, history
     
-    # Initialize conversation history state.
     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])
+    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)
-
+demo.launch(share=True, server_port=7862)