Update app.py

app.py

@@ -180,6 +180,9 @@ def chat_with_file(github_url: str, file_choice: int, user_query: str):
 ############################################
 # Gradio Interface Setup
 ############################################
+############################################
+# Gradio Interface Setup
+############################################
 
 with gr.Blocks() as demo:
     gr.Markdown("# RepoChat - Chat with Repository Files")
@@ -189,7 +192,8 @@ 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 = gr.Dropdown(label="Select a File", interactive=True)
+            # Set allow_custom_value=True to avoid errors when the selected value is not in the choices.
+            file_dropdown = gr.Dropdown(label="Select a File", interactive=True, allow_custom_value=True)
             repo_content_output = gr.Textbox(label="File Content", interactive=False, lines=10)
         with gr.Column(scale=2):
             gr.Markdown("### Chat Interface")
@@ -197,14 +201,16 @@ with gr.Blocks() as demo:
             chat_output = gr.Textbox(label="Chatbot Response", interactive=False, lines=10)
             chat_btn = gr.Button("Send Query")
     
-    # When clicking "Load Repository Contents", update file dropdown
+    # Function to update the file dropdown based on the GitHub repository URL.
     def update_file_dropdown(github_url):
         files = load_repo_contents(github_url)
+        if isinstance(files, str) and files.startswith("Error"):
+            return []
         return files
     
     load_repo_btn.click(fn=update_file_dropdown, inputs=[github_url_input], outputs=[file_dropdown])
     
-    # When file selection changes, update file content display
+    # Function to update the repository content display when a file is selected.
     def update_repo_content(github_url, file_choice):
         if not file_choice:
             return "No file selected."
@@ -217,10 +223,11 @@ with gr.Blocks() as demo:
     
     file_dropdown.change(fn=update_repo_content, inputs=[github_url_input, file_dropdown], outputs=[repo_content_output])
     
-    # When sending a chat query, process it
+    # Function to process the chat query.
     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()
+demo.launch(share=True)
+