Update app.py

app.py

@@ -69,7 +69,6 @@ class MyChatbot(gr.Chatbot):
             }
         ]
 
-# Update the Blocks implementation
 with gr.Blocks() as demo:
     with gr.Row():
         github_api_token = gr.Textbox(label="GitHub API Token", type="password")
@@ -108,36 +107,40 @@ with gr.Blocks() as demo:
     # Create state for storing conversation history
     state = gr.State([])
 
-    def user(user_message, history):
-        """Handle user messages"""
-        return "", history + [[user_message, None]]
+    # Add buttons with correct syntax
+    fetch_issues_button = gr.Button("Fetch Issues")
+    resolve_issue_button = gr.Button("Resolve Issue")
+    clear_button = gr.Button("Clear Chat")
+    submit_button = gr.Button("Submit Message")
 
-    def bot(history, system_msg, max_tokens, temp, top_p, api_token, username, repo, model, severity, lang):
-        """Generate bot response"""
-        if len(history) == 0:
-            return history
-            
-        user_message = history[-1][0]
-        response = respond(
-            command=user_message,
-            history=history[:-1],
-            system_message=system_msg,
-            max_tokens=max_tokens,
-            temperature=temp,
-            top_p=top_p,
-            github_api_token=api_token,
-            github_username=username,
-            github_repository=repo,
-            selected_model=model,
-            severity=severity,
-            programming_language=lang
-        )
-        
-        history[-1][1] = response['assistant_response']
-        return history
-
-    # Connect the components
-    msg.submit(
+    # Add a dropdown to select an issue
+    issue_dropdown = gr.Dropdown(
+        choices=[],
+        label="Select Issue",
+        interactive=True
+    )
+
+    def clear_chat():
+        return [], []
+
+    def fetch_issues(api_token, username, repo):
+        issues = fetch_github_issues(api_token, username, repo)
+        chatbot.issues = issues
+        return gr.Dropdown(choices=[f"{i+1}. {issue['title']}" for i, issue in enumerate(issues)])
+
+    # Connect button events
+    fetch_issues_button.click(
+        fetch_issues,
+        inputs=[github_api_token, github_username, github_repository],
+        outputs=[issue_dropdown]
+    )
+
+    clear_button.click(
+        clear_chat,
+        outputs=[chatbot, state]
+    )
+
+    submit_button.click(
         user,
         [msg, chatbot],
         [msg, chatbot]
@@ -159,56 +162,37 @@ with gr.Blocks() as demo:
         [chatbot]
     )
 
-    # Add a button to fetch GitHub issues
-    fetch_issues_button = gr.Button(label="Fetch Issues")
-    
-    def fetch_issues(api_token, username, repo):
-        """Fetch GitHub issues and update the chatbot"""
-        issues = fetch_github_issues(api_token, username, repo)
-        chatbot.issues = issues
-        return gr.Dropdown(choices=[f"{i+1}. {issue['title']}" for i, issue in enumerate(issues)])
-
-    fetch_issues_button.click(
-        fetch_issues,
-        inputs=[github_api_token, github_username, github_repository],
-        outputs=[issue_dropdown]
+    # Add message input handler
+    msg.submit(
+        user,
+        [msg, chatbot],
+        [msg, chatbot]
+    ).then(
+        bot,
+        [
+            chatbot,
+            system_message,
+            gr.Slider(minimum=1, maximum=8192, value=2048, step=1, label="Max new tokens"),
+            gr.Slider(minimum=0.1, maximum=4.0, value=0.71, step=0.1, label="Temperature"),
+            gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.01, label="Top-p"),
+            github_api_token,
+            github_username,
+            github_repository,
+            model_dropdown,
+            severity_dropdown,
+            programming_language_textbox
+        ],
+        [chatbot]
     )
 
-    # Add a dropdown to select an issue
-    issue_dropdown = gr.Dropdown(label="Select Issue", choices=[], interactive=True)
-    
-    def select_issue(selected, chatbot):
-        """Handle issue selection"""
-        if selected:
-            issue_num = int(selected.split('.')[0]) - 1
-            chatbot.current_issue = chatbot.issues[issue_num]
-            return chatbot.postprocess({
-                'command': str(issue_num + 1),
-                'github_api_token': github_api_token.value,
-                'github_username': github_username.value,
-                'github_repository': github_repository.value,
-                'selected_model': model_dropdown.value,
-                'severity': severity_dropdown.value,
-                'programming_language': programming_language_textbox.value
-            })
-        return []
-
+    # Issue selection handler
     issue_dropdown.change(
         select_issue,
         inputs=[issue_dropdown, chatbot],
         outputs=[chatbot]
     )
 
-    # Add a button to resolve an issue
-    resolve_issue_button = gr.Button(label="Resolve Issue")
-    
-    def resolve_selected_issue(api_token, username, repo, resolution):
-        """Resolve the currently selected issue"""
-        if chatbot.current_issue:
-            issue_number = chatbot.issues.index(chatbot.current_issue)
-            return resolve_issue(api_token, username, repo, issue_number, resolution)
-        return "No issue selected"
-
+    # Issue resolution handler
     resolve_issue_button.click(
         resolve_selected_issue,
         inputs=[