Update app.py

app.py

@@ -125,106 +125,11 @@ 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 Setup
+# Gradio Interface Functions
 ############################################
 
-# 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")
-#             # Dropdown with choices as file paths; default value is empty.
-#             file_dropdown = gr.Dropdown(label="Select a File", interactive=True, value="", choices=[])
-#             # repo_content_output = gr.Textbox(label="File Content", interactive=False, lines=30)
-#             repo_content_output = gr.Chatbot(label="Chat Conversation")
-#         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")
-    
-#     # Callback: Update file dropdown choices.
-#     def update_file_dropdown(github_url):
-#         files = load_repo_contents_backend(github_url)
-#         if isinstance(files, str):  # Error message
-#             print("Error loading files:", files)
-#             return gr.update(choices=[], value="")
-#         print("Files loaded:", files)
-#         # Do not pre-select any file (empty value)
-#         return gr.update(choices=files, value="")
-    
-#     load_repo_btn.click(fn=update_file_dropdown, inputs=[github_url_input], outputs=[file_dropdown])
-    
-#     # Callback: Update repository content when a file is selected.
-#     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.
-#     def process_chat(github_url, file_choice, chat_query):
-#         if not file_choice:
-#             return "Please select a file first."
-#         return chat_with_file(github_url, 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)
-
-
+# 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)
@@ -234,16 +139,25 @@ def get_file_content_for_choice(github_url: str, file_path: str):
     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[:1000]  # Use the first 1000 characters as context
+    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)
@@ -252,61 +166,71 @@ 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 not isinstance(tree_data, dict) or "tree" not in tree_data:
+    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
-############################################
+###########################################
+Gradio Interface Setup
+###########################################
 
 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, choices=[])
-            repo_content_output = gr.Textbox(label="File Content", interactive=False, lines=30)
-            # repo_content_output = gr.Chatbot(label="File Content")
-        
+            # Dropdown with choices as file paths; default value is empty.
+            file_dropdown = gr.Dropdown(label="Select a File", interactive=True, value="", choices=[])
+            # repo_content_output = gr.Textbox(label="File Content", interactive=False, lines=30)
+            repo_content_output = gr.Chatbot(label="Chat Conversation")
         with gr.Column(scale=2):
             gr.Markdown("### Chat Interface")
-            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")
-
-    # State to hold conversation history
-    conversation_history = gr.State([])
+            # chat_output = gr.Textbox(label="Chatbot Response", interactive=False, lines=10)
+            chat_output = gr.Chatbot(label="File Content")
 
+            chat_btn = gr.Button("Send Query")
+    
+    # Callback: Update file dropdown choices.
     def update_file_dropdown(github_url):
         files = load_repo_contents_backend(github_url)
         if isinstance(files, str):  # Error message
-            return gr.update(choices=[]), files
-        return gr.update(choices=files), ""
+            print("Error loading files:", files)
+            return gr.update(choices=[], value="")
+        print("Files loaded:", files)
+        # Do not pre-select any file (empty value)
+        return gr.update(choices=files, value="")
+    
+    load_repo_btn.click(fn=update_file_dropdown, inputs=[github_url_input], outputs=[file_dropdown])
+    
+    # Callback: Update repository content when a file is selected.
+    # 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
 
     def update_repo_content(github_url, file_choice):
         if not file_choice:
-            return "No file selected."
+            return [("System", "No file selected.")]
         content, _ = get_file_content_for_choice(github_url, file_choice)
-        return content
+    # Wrap the content in a tuple so it displays like a chat message.
+        return [("File Content", content)]
 
     
-
-
-    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
-
-    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])
+    
+    # Callback: Process chat query.
+    def process_chat(github_url, file_choice, chat_query):
+        if not file_choice:
+            return "Please select a file first."
+        return chat_with_file(github_url, 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)
 
-demo.launch(server_port=7862)