Update app.py

app.py

@@ -136,51 +136,70 @@ def get_llm_response(prompt: str, model_name: str = "meta-llama/Llama-2-7b-chat-
 # Gradio Interface Functions
 ############################################
 
+
+# Debug: Ensure load_repo_contents returns the file list properly.
 def load_repo_contents(github_url: str):
     try:
-        owner, repo = extract_repo_info(github_url)
+        files = load_repo_contents_backend(github_url)
+        print("Files loaded:", files)  # Debug print
+        return files
     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
+        print("Error loading repo contents:", str(e))
+        return []
 
-def get_file_content_for_choice(github_url: str, file_choice: int):
+# Original function renamed to avoid conflict.
+def load_repo_contents_backend(github_url: str):
     try:
         owner, repo = extract_repo_info(github_url)
     except Exception as e:
-        return str(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"]
-    if file_choice < 1 or file_choice > len(file_list):
-        return "Error: Invalid file choice."
-    selected_file = file_list[file_choice - 1]
-    content = get_file_content(owner, repo, selected_file)
-    return content, selected_file
-
-
-def chat_with_file(github_url: str, file_choice: int, user_query: str):
-    result = get_file_content_for_choice(github_url, file_choice)
-    if isinstance(result, str):
-        return result  # Error message
-    file_content, selected_file = result
-    preprocessed = preprocess_text(file_content)
-    context_snippet = preprocessed[:1000]  # use first 1000 characters as context
-    prompt = generate_prompt(user_query, [context_snippet])
-    llm_response = get_llm_response(prompt)
-    return f"File: {selected_file}\n\nLLM Response:\n{llm_response}"
+    return file_list
 
-############################################
-# Gradio Interface Setup
-############################################
+# Update the file dropdown without allow_custom_value.
+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 = gr.Dropdown(label="Select a File", interactive=True)  # Removed allow_custom_value
+            repo_content_output = gr.Textbox(label="File Content", interactive=False, lines=10)
+        with gr.Column(scale=2):
+            gr.Markdown("### Chat Interface")
+            chat_query_input = gr.Textbox(label="Your Query", placeholder="Type your query here")
+            chat_output = gr.Textbox(label="Chatbot Response", interactive=False, lines=10)
+            chat_btn = gr.Button("Send Query")
+    
+    # Update file dropdown based on GitHub URL.
+    load_repo_btn.click(fn=load_repo_contents, inputs=[github_url_input], outputs=[file_dropdown])
+    
+    def update_repo_content(github_url, file_choice):
+        if not file_choice:
+            return "No file selected."
+        try:
+            file_index = int(file_choice)
+        except Exception as e:
+            print("Error converting file choice:", str(e))
+            file_index = 1
+        content, _ = get_file_content_for_choice(github_url, file_index)
+        return content
+    
+    file_dropdown.change(fn=update_repo_content, inputs=[github_url_input, file_dropdown], outputs=[repo_content_output])
+    
+    def process_chat(github_url, file_choice, chat_query):
+        return chat_with_file(github_url, int(file_choice), chat_query)
+    
+    chat_btn.click(fn=process_chat, inputs=[github_url_input, file_dropdown, chat_query_input], outputs=[chat_output])
+    
+demo.launch(share=True)
 ############################################
 # Gradio Interface Setup
 ############################################