Update app.py

app.py

@@ -9,62 +9,26 @@ import textwrap
 import google.generativeai as genai
 import asyncio
 from typing import Generator, AsyncGenerator
+from openai import AsyncOpenAI
+import dotenv
+
+# Load environment variables
+dotenv.load_dotenv()
 
 # Metadata
-CURRENT_TIME = "2025-05-23 12:44:47"
+CURRENT_TIME = "2025-05-23 12:57:22"
 CURRENT_USER = "ErRickow"
 
-# Link bantuan
-GITHUB_TOKEN_HELP = """
-### Cara Mendapatkan GitHub Token:
-
-1. Kunjungi [GitHub Token Settings](https://github.com/settings/tokens)
-2. Klik "Generate new token" > "Generate new token (classic)"
-3. Beri nama token Anda di "Note"
-4. Pilih scope:
-   - `repo` (untuk akses repository private)
-   - `read:packages` (opsional, untuk akses package)
-5. Klik "Generate token"
-6. **PENTING**: Salin token segera! Token hanya ditampilkan sekali
-
-Token diperlukan untuk:
-- Mengakses repository private
-- Clone repository dengan rate limit lebih tinggi
-- Mengakses fitur GitHub API
-"""
-
-GEMINI_API_HELP = """
-### Cara Mendapatkan Gemini API Key:
-
-1. Kunjungi [Google AI Studio](https://makersuite.google.com/app/apikey)
-2. Login dengan akun Google Anda
-3. Klik "Create API Key"
-4. Salin API Key yang dihasilkan
-
-Catatan:
-- Gemini memberikan kuota gratis setiap bulan
-- Key bisa dibuat ulang jika diperlukan
-- Monitor penggunaan di [Google Cloud Console](https://console.cloud.google.com/)
-"""
-
-OLLAMA_HELP = """
-### Cara Menggunakan Ollama:
-
-1. Install Ollama dari [ollama.ai](https://ollama.ai)
-2. Jalankan Ollama di komputer Anda
-3. Pastikan Ollama berjalan di http://localhost:11434
-
-Catatan:
-- Ollama berjalan secara lokal di komputer Anda
-- Tidak memerlukan API key
-- Ideal untuk privasi dan penggunaan offline
-"""
+# Default API Keys (fallback if user doesn't provide their own)
+DEFAULT_XAI_KEY = os.getenv("XAI_API_KEY", "xai-vfjhklL384Z4HKdItsZomqpFlXubTZJAFnISQUpV7dE8lRnWwYBVPSCxSTlu08wDbAcv720bx2dDiQ9x")
+DEFAULT_GEMINI_KEY = os.getenv("GEMINI_API_KEY")
 
 # API settings
 OLLAMA_API = os.environ.get("OLLAMA_API", "http://localhost:11434")
+XAI_BASE_URL = "https://api.x.ai/v1"
 
 # Model lists
-DEFAULT_MODELS = [
+OLLAMA_MODELS = [
     "llama2",
     "codellama",
     "mistral",
@@ -75,9 +39,34 @@ DEFAULT_MODELS = [
     "orca-mini"
 ]
 
+XAI_MODELS = [
+    "grok-2-latest",
+    "grok-1",
+]
+
+GEMINI_MODELS = [
+    "gemini-1.5-mini",
+    "gemini-pro-vision",
+]
+
+# Help texts
+XAI_API_HELP = """
+### Cara Mendapatkan X.AI (Grok) API Key:
+
+1. Kunjungi [X.AI Developer Portal](https://x.ai)
+2. Daftar/Login ke akun Anda
+3. Buat API Key baru
+4. Salin API Key
+
+Note:
+- Jika tidak diisi, akan menggunakan API key default
+- Masukkan API key Anda sendiri jika default mencapai limit
+"""
+
 class AIProvider:
     OLLAMA = "ollama"
     GEMINI = "gemini"
+    XAI = "xai"
 
 class RepoAnalyzer:
     def __init__(self):
@@ -85,6 +74,55 @@ class RepoAnalyzer:
         self.repo_content = {}
         self.chat_history = []
 
+    async def stream_xai_response(self, prompt: str, api_key: str = None, model: str = "grok-2-latest") -> AsyncGenerator[str, None]:
+        """Stream response dari X.AI (Grok) API"""
+        try:
+            # Use default key if none provided
+            actual_key = api_key if api_key else DEFAULT_XAI_KEY
+            
+            if not actual_key:
+                yield "⚠️ API Key X.AI diperlukan. Gunakan key Anda sendiri atau tunggu reset limit default key."
+                return
+
+            client = AsyncOpenAI(
+                api_key=actual_key,
+                base_url=XAI_BASE_URL
+            )
+
+            # Prepare messages with repository context if available
+            messages = [
+                {"role": "system", "content": "Anda adalah asisten AI yang membantu menganalisis repository code. Berikan respons dalam Bahasa Indonesia."}
+            ]
+            
+            if self.current_repo:
+                context = f"Repository: {self.current_repo}\n\n"
+                repo_files = "\n".join(list(self.repo_content.keys()))
+                context += f"Files in repository:\n{repo_files}\n\n"
+                messages.append({"role": "system", "content": context})
+
+            messages.append({"role": "user", "content": prompt})
+
+            stream = await client.chat.completions.create(
+                model=model,
+                messages=messages,
+                stream=True
+            )
+
+            full_response = ""
+            async for chunk in stream:
+                if chunk.choices[0].delta.content:
+                    content = chunk.choices[0].delta.content
+                    full_response += content
+                    yield content
+
+            self.chat_history.append({"role": "user", "content": prompt})
+            self.chat_history.append({"role": "assistant", "content": full_response})
+
+        except Exception as e:
+            error_msg = f"⚠️ Error dalam X.AI API: {str(e)}"
+            print(error_msg)
+            yield error_msg
+
     async def stream_gemini_response(self, prompt: str, api_key: str) -> AsyncGenerator[str, None]:
         """Stream response dari Gemini API"""
         try:
@@ -93,7 +131,7 @@ class RepoAnalyzer:
                 return
 
             genai.configure(api_key=api_key)
-            model = genai.GenerativeModel('gemini-1.5-mini')
+            model = genai.GenerativeModel('gemini-pro')
             
             # Tambahkan konteks repository jika ada
             if self.current_repo:
@@ -207,21 +245,53 @@ def create_ui():
 
         with gr.Tab("🛠️ Konfigurasi"):
             provider = gr.Radio(
-                choices=[AIProvider.GEMINI, AIProvider.OLLAMA],
+                choices=[AIProvider.XAI, AIProvider.GEMINI, AIProvider.OLLAMA],
                 label="Penyedia AI",
-                value=AIProvider.GEMINI
+                value=AIProvider.XAI
             )
             
             with gr.Group() as api_settings:
+                with gr.Row():
+                    xai_key = gr.Textbox(
+                        label="X.AI (Grok) API Key",
+                        type="password",
+                        placeholder="Opsional - Klik icon (?) untuk info. Kosongkan untuk gunakan key default",
+                        show_label=True
+                    )
+                    gr.Markdown(XAI_API_HELP)
+                
                 with gr.Row():
                     gemini_key = gr.Textbox(
                         label="Gemini API Key",
                         type="password",
-                        placeholder="Klik icon (?) untuk panduan mendapatkan API key",
+                        placeholder="Opsional - Kosongkan untuk gunakan key default",
                         show_label=True
                     )
                     gr.Markdown(GEMINI_API_HELP)
 
+            # Model selection based on provider
+            with gr.Row():
+                model_dropdown = gr.Dropdown(
+                    label="Model AI",
+                    choices=XAI_MODELS,
+                    value="grok-2-latest",
+                    interactive=True
+                )
+
+            def update_model_list(provider_choice):
+                if provider_choice == AIProvider.XAI:
+                    return gr.Dropdown(choices=XAI_MODELS, value="grok-2-latest")
+                elif provider_choice == AIProvider.GEMINI:
+                    return gr.Dropdown(choices=GEMINI_MODELS, value="gemini-pro")
+                else:  # OLLAMA
+                    return gr.Dropdown(choices=OLLAMA_MODELS, value="llama2")
+            
+            provider.change(
+                fn=update_model_list,
+                inputs=[provider],
+                outputs=[model_dropdown]
+            )
+
         with gr.Tab("📊 Analisis Repository"):
             with gr.Row():
                 repo_url = gr.Textbox(
@@ -267,7 +337,7 @@ def create_ui():
                 show_label=True
             )
 
-            async def handle_chat(message, history, api_key):
+            async def handle_chat(message, history, provider_choice, model_name, xai_key, gemini_key):
                 if not analyzer.current_repo:
                     return history + [[message, "⚠️ Mohon clone repository terlebih dahulu sebelum mengajukan pertanyaan."]]
                 
@@ -275,24 +345,31 @@ def create_ui():
                 history.append([message, ""])
                 
                 full_response = ""
-                async for chunk in analyzer.stream_gemini_response(message, api_key):
-                    full_response += chunk
-                    history[-1][1] = full_response
+                if provider_choice == AIProvider.XAI:
+                    async for chunk in analyzer.stream_xai_response(message, xai_key, model_name):
+                        full_response += chunk
+                        history[-1][1] = full_response
+                        yield history
+                elif provider_choice == AIProvider.GEMINI:
+                    async for chunk in analyzer.stream_gemini_response(message, gemini_key or DEFAULT_GEMINI_KEY):
+                        full_response += chunk
+                        history[-1][1] = full_response
+                        yield history
+                else:  # OLLAMA
+                    response = analyze_with_ollama(model_name, message)
+                    history[-1][1] = response
                     yield history
 
-            def handle_clone(url, token, branch_name):
-                success, result = analyzer.clone_repository(url, token, branch_name if branch_name else None)
-                return result
-
-            clone_button.click(
-                fn=handle_clone,
-                inputs=[repo_url, github_token, branch],
-                outputs=clone_status
-            )
-
             send_button.click(
                 fn=handle_chat,
-                inputs=[chat_input, chat_history, gemini_key],
+                inputs=[
+                    chat_input,
+                    chat_history,
+                    provider,
+                    model_dropdown,
+                    xai_key,
+                    gemini_key
+                ],
                 outputs=chat_history
             ).then(
                 fn=lambda: gr.update(value=""),
@@ -301,7 +378,14 @@ def create_ui():
 
             chat_input.submit(
                 fn=handle_chat,
-                inputs=[chat_input, chat_history, gemini_key],
+                inputs=[
+                    chat_input,
+                    chat_history,
+                    provider,
+                    model_dropdown,
+                    xai_key,
+                    gemini_key
+                ],
                 outputs=chat_history
             ).then(
                 fn=lambda: gr.update(value=""),