Update app.py

app.py

@@ -19,25 +19,22 @@ class ChatInterface:
         self.image_client = InferenceClient(image_model, token=hf_token)
         self.custom_responses = self._initialize_custom_responses()
         self.system_prompt = self._initialize_system_prompt()
-        
+
     @staticmethod
     def _initialize_system_prompt() -> str:
         """Initialize the system prompt for the AI assistant."""
         return """# Xylaria AI Assistant (v1.3.0)
-
 ## Core Identity
 - Name: Xylaria
 - Version: 1.3.0
 - Base Model: Mistral-Nemo-Instruct
 - Knowledge Cutoff: April 2024
-
 ## Primary Directives
 1. Provide accurate, well-researched information
 2. Maintain ethical standards in all interactions
 3. Adapt communication style to user needs
 4. Acknowledge limitations and uncertainties
 5. Prioritize user safety and wellbeing
-
 ## Technical Capabilities
 - Programming & Software Development
 - Mathematical Analysis & Computation
@@ -46,39 +43,33 @@ class ChatInterface:
 - Technical Writing & Documentation
 - Problem-Solving & Debugging
 - Educational Content Creation
-
 ## Communication Guidelines
 - Use clear, precise language
 - Adapt technical depth to user expertise
 - Provide step-by-step explanations when needed
 - Ask for clarification when necessary
 - Maintain professional yet approachable tone
-
 ## Domain Expertise
 1. Computer Science & Technology
    - Multiple programming languages
    - Software architecture & design
    - Data structures & algorithms
    - Best practices & patterns
-
 2. Mathematics & Statistics
    - Advanced mathematical concepts
    - Statistical analysis
    - Probability theory
    - Data interpretation
-
 3. Sciences
    - Physics & Chemistry
    - Biology & Life Sciences
    - Environmental Science
    - Engineering Principles
-
 4. Humanities & Arts
    - Technical Writing
    - Documentation
    - Creative Problem-Solving
    - Research Methodology
-
 ## Response Framework
 1. Analyze user query thoroughly
 2. Consider context and background
@@ -86,7 +77,6 @@ class ChatInterface:
 4. Provide examples when helpful
 5. Verify accuracy of information
 6. Include relevant caveats or limitations
-
 ## Ethical Guidelines
 - Prioritize user safety
 - Maintain data privacy
@@ -94,14 +84,12 @@ class ChatInterface:
 - Acknowledge uncertainties
 - Provide balanced perspectives
 - Respect intellectual property
-
 ## Limitations
 - No real-time data access
 - No persistent memory between sessions
 - Cannot verify external sources
 - No capability to execute code
 - Limited to text and basic image generation
-
 ## Version-Specific Features
 - Enhanced error handling
 - Improved response consistency
@@ -117,7 +105,7 @@ class ChatInterface:
             "developer": ["sk md saad amin"],
             "strawberry_r": ["3"]
         }
-        
+
         patterns = {}
         name_variations = [
             "what is ur name", "what's ur name", "whats ur name", 
@@ -131,19 +119,19 @@ class ChatInterface:
             "how many 'r' is in strawberry", "how many r is in strawberry",
             "how many r's are in strawberry"
         ]
-        
+
         for pattern in name_variations:
             patterns[pattern] = "xylaria"
             patterns[pattern.capitalize()] = "xylaria"
-        
+
         for pattern in dev_variations:
             patterns[pattern] = "sk md saad amin"
             patterns[pattern.capitalize()] = "sk md saad amin"
-            
+
         for pattern in strawberry_variations:
             patterns[pattern] = "3"
             patterns[pattern.capitalize()] = "3"
-            
+
         return patterns
 
     @retry(
@@ -165,8 +153,7 @@ class ChatInterface:
                 max_tokens=max_tokens,
                 stream=True,
                 temperature=temperature,
-                top_p=top_p,
-                timeout=30
+                top_p=top_p
             ):
                 token = message.choices[0].delta.content
                 response += token
@@ -191,8 +178,7 @@ class ChatInterface:
                     "sampling_steps": 15,
                     "upscaler": "4x-UltraSharp",
                     "denoising_strength": 0.5,
-                },
-                timeout=60
+                }
             )
         except Exception as e:
             logger.error(f"Error generating image: {e}")
@@ -260,7 +246,7 @@ def create_interface(hf_token: str):
         image_model="SG161222/RealVisXL_V3.0",
         hf_token=hf_token
     )
-    
+
     return gr.ChatInterface(
         partial(chat.respond),
         additional_inputs=[
@@ -288,18 +274,13 @@ def create_interface(hf_token: str):
         ],
         css="""
         @import url('https://fonts.googleapis.com/css2?family=Inter:wght@400;600&display=swap');
-        body, .gradio-container {
+        body {
             font-family: 'Inter', sans-serif;
         }
         """
     )
 
 if __name__ == "__main__":
-    # Get token from environment variable
-    hf_token = os.getenv("hf_token")
-    if not hf_token:
-        raise ValueError("Please set the hf_token environment variable")
-    
-    # Create and launch the interface
-    demo = create_interface(hf_token)
-    demo.launch()
+    # Replace with your actual Hugging Face token
+    hf_token = "your_hf_token"
+    gr.Interface(fn=create_interface(hf_token), live=True).launch()