Update app.py

app.py

@@ -5,6 +5,8 @@ import gradio as gr
 from datetime import datetime
 from typing import List, Dict, Any, Optional, Union
 import threading
+
+# Import ZeroGPU spaces
 import spaces
 
 # Import Groq
@@ -30,7 +32,8 @@ class CreativeAgenticAI:
         self.client = Groq(api_key=self.api_key)
         self.model = model
         self.conversation_history = []
-        
+    
+    @spaces.GPU
     def chat(self, message: str, 
              include_domains: List[str] = None,
              exclude_domains: List[str] = None,
@@ -158,6 +161,7 @@ class CreativeAgenticAI:
 ai_instance = None
 api_key_status = "Not Set"
 
+@spaces.GPU
 def validate_api_key(api_key: str, model: str) -> str:
     """Validate Groq API key and initialize AI instance"""
     global ai_instance, api_key_status
@@ -197,6 +201,7 @@ def update_model(model: str) -> str:
     else:
         return "⚠️ Please set your API key first"
 
+@spaces.GPU
 def chat_with_ai(message: str, 
                 include_domains: str, 
                 exclude_domains: str,
@@ -304,7 +309,7 @@ def create_gradio_app():
         gr.HTML("""
         <div class="header">
             <h1>🤖 NeuroScope-AI</h1>
-            <p>Powered by Groq's Compound Models with Web Search & Agentic Capabilities</p>
+            <p>Powered by Groq's Compound Models with Web Search & Agentic Capabilities | Enhanced with ZeroGPU</p>
         </div>
         """)
 
@@ -316,6 +321,7 @@ def create_gradio_app():
                     - 🔍 Domain filtering (Scope)
                     - 🤖 AI capabilities (AI)
                     - ⚡ Precision (Scope implies focused, targeted results)
+                    - 🚀 Enhanced with ZeroGPU for optimized performance
                 """)
         with gr.Group():
             with gr.Accordion("📚 IMPORTANT !!!", open=False, elem_id="neuroscope-accordion"):
@@ -350,15 +356,25 @@ def create_gradio_app():
                 
                 ### 🎭 **Custom System Prompt Feature**
                 
-                Allows complete override of the AI’s **default personality and behavior**.
+                Allows complete override of the AI's **default personality and behavior**.
                 You can redefine the AI to act as:
                 
                   - A **professional business consultant**
                   - A **coding mentor**
                   - A **creative writer**
                   - A **specific character or persona**
-                - Provides full control to **reshape the AI’s tone, expertise, and conversational style** with a single prompt.
-
+                - Provides full control to **reshape the AI's tone, expertise, and conversational style** with a single prompt.
+                
+                ---
+                
+                ### 🚀 **ZeroGPU Integration**
+                
+                This application is enhanced with ZeroGPU for:
+                
+                  - **Optimized GPU utilization** for AI inference
+                  - **Improved response times** and performance
+                  - **Efficient resource management** in cloud environments
+                  - **Better scalability** for multiple concurrent users
                 """)
         
         # API Key Section
@@ -499,12 +515,14 @@ def create_gradio_app():
             - **Creative Responses**: Optimized for engaging and helpful conversations
             - **Memory**: Maintains conversation context throughout the session
             - **Customizable**: Adjust temperature, tokens, and system prompts
+            - **ZeroGPU Enhanced**: Optimized performance with GPU acceleration
             
             ### 💡 Tips for Best Results
             - Be specific in your questions for better responses
             - Use domain filtering for specialized research
             - Adjust temperature: higher for creativity, lower for precision
             - Try different system prompts for different conversation styles
+            - ZeroGPU ensures faster processing for complex queries
             """)
         
         # Sample Examples Section
@@ -570,8 +588,8 @@ def create_gradio_app():
         with gr.Accordion("🚀 About This Tool", open=True, elem_id="neuroscope-accordion"):
             gr.Markdown("""
             A Creative Agentic AI Chat Tool leverages Groq's powerful compound models to provide intelligent, 
-            context-aware responses with web search capabilities. Perfect for research, programming help, 
-            creative writing, and complex problem-solving.
+            context-aware responses with web search capabilities. Enhanced with ZeroGPU for optimal performance.
+            Perfect for research, programming help, creative writing, and complex problem-solving.
             
             **Features:**
             - 🔍 Web search with domain filtering
@@ -579,17 +597,15 @@ def create_gradio_app():
             - 💬 Conversational memory and context
             - ⚙️ Customizable parameters and prompts
             - 🎨 Creative and analytical capabilities
+            - 🚀 ZeroGPU acceleration for faster processing
+            - ⚡ Optimized resource utilization
             """)
     
     return app
 
-# Main execution with ZeroGPU
-@spaces.GPU
-def main():
+# Main execution
+if __name__ == "__main__":
     app = create_gradio_app()
     app.launch(
         share=True
-    )
-
-if __name__ == "__main__":
-    main()
+    )