Update app.py

app.py

@@ -136,9 +136,6 @@ def get_llm_response(prompt: str, model_name: str = "meta-llama/Llama-2-7b-chat-
 # Gradio Interface Functions
 ############################################
 
-############################################
-# Gradio Interface Setup
-############################################
 
 with gr.Blocks() as demo:
     gr.Markdown("# RepoChat - Chat with Repository Files")
@@ -148,7 +145,7 @@ 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")
-            # Set default value to empty and do not allow custom values.
+            # Initialize dropdown with empty choices and an empty default value.
             file_dropdown = gr.Dropdown(label="Select a File", interactive=True, value="", choices=[])
             repo_content_output = gr.Textbox(label="File Content", interactive=False, lines=10)
         with gr.Column(scale=2):
@@ -157,7 +154,7 @@ with gr.Blocks() as demo:
             chat_output = gr.Textbox(label="Chatbot Response", interactive=False, lines=10)
             chat_btn = gr.Button("Send Query")
     
-    # Function to fetch repository files from GitHub.
+    # Function to load repository contents from GitHub.
     def load_repo_contents_backend(github_url: str):
         try:
             owner, repo = extract_repo_info(github_url)
@@ -171,39 +168,39 @@ with gr.Blocks() as demo:
         file_list = [item["path"] for item in tree_data["tree"] if item["type"] == "blob"]
         return file_list
 
+    # Callback to update the file dropdown.
     def update_file_dropdown(github_url):
         files = load_repo_contents_backend(github_url)
-        # Check if files is a string (an error message) or a list.
-        if isinstance(files, str):
+        if isinstance(files, str):  # If there's an error message.
             print("Error loading files:", files)
-            return []  # Return an empty list on error.
+            return gr.Dropdown.update(choices=[], value="")
         print("Files loaded:", files)
-        return files
-    
+        # Set the choices but leave value as empty to force user selection.
+        return gr.Dropdown.update(choices=files, value="")
+
     load_repo_btn.click(fn=update_file_dropdown, inputs=[github_url_input], outputs=[file_dropdown])
     
-    # Function to update the repository content display based on the selected file.
+    # Callback to update the repository content display based on the selected file.
     def update_repo_content(github_url, file_choice):
-        if not file_choice or 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 to int:", str(e))
+            print("Error converting file choice:", str(e))
             return "Invalid file selection."
         content_tuple = get_file_content_for_choice(github_url, file_index)
         if isinstance(content_tuple, str):
-            # An error occurred
+            # Return error message if one occurred.
             return content_tuple
         content, _ = content_tuple
         return content
-    
+
     file_dropdown.change(fn=update_repo_content, inputs=[github_url_input, file_dropdown], outputs=[repo_content_output])
     
-    # Function to process the chat query.
+    # Callback to process the chat query.
     def process_chat(github_url, file_choice, chat_query):
-        # Ensure file_choice is not empty.
-        if not file_choice or file_choice == "":
+        if not file_choice:
             return "Please select a file first."
         try:
             file_index = int(file_choice)