Update appaaa.py

appaaa.py

@@ -352,33 +352,181 @@ async def respond(
         # Handle other commands or provide a default response
         yield "I'm not sure what you mean. Try using `/help` for a list of available commands."
 
-# Gradio Interface
-iface = gr.Interface(
-    fn=respond,
-    inputs=[
-        gr.Textbox(label="Command", lines=1),
-        gr.State(label="Chat History", value=[]),
-        gr.Textbox(label="System Message", lines=1, value="You are a helpful and informative AI assistant."),
-        gr.Slider(label="Max Tokens", minimum=1, maximum=1000, value=500, step=1),
-        gr.Slider(label="Temperature", minimum=0.0, maximum=1.0, value=0.7, step=0.1),
-        gr.Slider(label="Top-p", minimum=0.0, maximum=1.0, value=0.9, step=0.1),
-        gr.Textbox(label="GitHub API Token", lines=1, placeholder="Enter your GitHub API token"),
-        gr.Textbox(label="GitHub Username", lines=1, placeholder="Enter your GitHub username"),
-        gr.Textbox(label="GitHub Repository", lines=1, placeholder="Enter your GitHub repository"),
-        gr.Dropdown(label="Model", choices=["text-davinci-003", "text-curie-001", "text-babbage-001", "text-ada-001"], value="text-davinci-003"),
-        gr.Dropdown(label="Severity", choices=["Low", "Medium", "High"], value="Medium"),
-        gr.Dropdown(label="Programming Language", choices=["Python", "JavaScript", "Java", "C++", "C#"], value="Python"),
-    ],
-    outputs=gr.Textbox(label="Response"),
-    title="AI Assistant",
-    description="Ask me anything, or use commands to interact with GitHub.",
-    examples=[
-        "/generate Create a REST API with FastAPI",
-        "/analyze Memory leak in data processing pipeline",
-        "/github octocat hello-world ghp_12345",
-        "/list_issues"
-    ],
-    inputs=[chat.textbox]  # Point to chat input instead of model selector
-)
-
-iface.launch(share=True)
+
+
+def create_gradio_interface():
+    def process_command(
+        command: str,
+        system_message: str,
+        max_tokens: int,
+        temperature: float,
+        top_p: float,
+        github_token: str,
+        github_username: str,
+        github_repo: str,
+        model: str,
+        severity: str,
+        programming_language: str
+    ):
+        try:
+            # Convert the synchronous call to async
+            import asyncio
+            return asyncio.run(respond(
+                command=command,
+                history=[],
+                system_message=system_message,
+                max_tokens=max_tokens,
+                temperature=temperature,
+                top_p=top_p,
+                github_api_token=github_token,
+                github_username=github_username,
+                github_repository=github_repo,
+                selected_model=model,
+                severity=severity,
+                programming_language=programming_language
+            ))
+        except Exception as e:
+            return f"Error: {str(e)}"
+
+    # Create the Gradio interface
+    with gr.Blocks(title="AI Assistant") as demo:
+        gr.Markdown("""
+        # AI Assistant
+        Ask me anything, or use commands to interact with GitHub.
+        
+        Available commands:
+        - `/github <username> <repo> <token>`: Connect to GitHub
+        - `/help`: Show help
+        - `/generate_code`: Generate code
+        - `/analyze`: Analyze issues
+        - `/list_issues`: List repository issues
+        """)
+
+        with gr.Row():
+            with gr.Column():
+                command_input = gr.Textbox(
+                    label="Command",
+                    placeholder="Enter command (e.g., /help)",
+                    lines=2
+                )
+                
+                system_message = gr.Textbox(
+                    label="System Message",
+                    value="You are a helpful AI assistant.",
+                    lines=2
+                )
+
+            with gr.Column():
+                github_token = gr.Textbox(
+                    label="GitHub Token",
+                    type="password",
+                    placeholder="Enter GitHub token"
+                )
+                github_username = gr.Textbox(
+                    label="GitHub Username",
+                    placeholder="Enter GitHub username"
+                )
+                github_repo = gr.Textbox(
+                    label="GitHub Repository",
+                    placeholder="Enter repository name"
+                )
+
+        with gr.Row():
+            with gr.Column():
+                model = gr.Dropdown(
+                    label="Model",
+                    choices=["zephyr-7b-beta"],
+                    value="zephyr-7b-beta"
+                )
+                severity = gr.Dropdown(
+                    label="Severity",
+                    choices=["Low", "Medium", "High"],
+                    value="Medium"
+                )
+                programming_language = gr.Dropdown(
+                    label="Programming Language",
+                    choices=["Python", "JavaScript", "Java", "C++", "C#"],
+                    value="Python"
+                )
+
+            with gr.Column():
+                max_tokens = gr.Slider(
+                    label="Max Tokens",
+                    minimum=50,
+                    maximum=1000,
+                    value=500,
+                    step=50
+                )
+                temperature = gr.Slider(
+                    label="Temperature",
+                    minimum=0.1,
+                    maximum=1.0,
+                    value=0.7,
+                    step=0.1
+                )
+                top_p = gr.Slider(
+                    label="Top-p",
+                    minimum=0.1,
+                    maximum=1.0,
+                    value=0.9,
+                    step=0.1
+                )
+
+        submit_btn = gr.Button("Submit")
+        response_output = gr.Textbox(
+            label="Response",
+            lines=10,
+            placeholder="Response will appear here..."
+        )
+
+        # Handle submit button click
+        submit_btn.click(
+            fn=process_command,
+            inputs=[
+                command_input,
+                system_message,
+                max_tokens,
+                temperature,
+                top_p,
+                github_token,
+                github_username,
+                github_repo,
+                model,
+                severity,
+                programming_language
+            ],
+            outputs=response_output
+        )
+
+        # Add example commands
+        gr.Examples(
+            examples=[
+                ["/help", "You are a helpful AI assistant.", 500, 0.7, 0.9, "", "", "", "zephyr-7b-beta", "Medium", "Python"],
+                ["/github octocat hello-world YOUR_TOKEN", "You are a helpful AI assistant.", 500, 0.7, 0.9, "", "", "", "zephyr-7b-beta", "Medium", "Python"],
+                ["/generate_code Create a FastAPI REST API", "You are a helpful AI assistant.", 500, 0.7, 0.9, "", "", "", "zephyr-7b-beta", "Medium", "Python"],
+            ],
+            inputs=[
+                command_input,
+                system_message,
+                max_tokens,
+                temperature,
+                top_p,
+                github_token,
+                github_username,
+                github_repo,
+                model,
+                severity,
+                programming_language
+            ]
+        )
+
+    return demo
+
+# Launch the interface
+if __name__ == "__main__":
+    demo = create_gradio_interface()
+    demo.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        share=False
+    )