Update app.py

app.py

@@ -1,3 +1,5 @@
+modify the code to allow the send button to work with enter and make other necessary optimizations.
+
 import os
 import re
 import cohere
@@ -6,7 +8,6 @@ from typing import Iterator, List, Tuple
 
 # Configuration Constants
 DEFAULT_SYSTEM_PROMPT = """ 
-  
 أنت مترجم ثنائي اللغة متخصص في الترجمة بين العربية والإنجليزية. هدفك هو تقديم ترجمات دقيقة، ملائمة للسياق، ومتسقة من الناحية الأسلوبية، مع الالتزام بالإرشادات التالية:
 أسلوب الكتابة:
 1. الدقة النحوية: احرص دائمًا على أن تكون الترجمة صحيحة نحويًا.
@@ -39,17 +40,13 @@ DEFAULT_SYSTEM_PROMPT = """
 حافظ على تنسيق البيانات المهمة (مثل التواريخ، والقياسات، والاستشهادات القانونية).
 عند الشك:
 قدم الأولوية للوضوح، والتناسق، والملاءمة مع احتياجات الجمهور المستهدف. قم دائمًا بموازنة التعليمات الخاصة بالمشروع مع هذه الإرشادات، مع إعطاء الأولوية لمتطلبات العميل عند وجود أي تعارض.
- 
-
-
-"""  # Full prompt maintained as original
+"""  # (keeping the full prompt as in original)
 
 TITLE = "<h1><center>Mawared T Assistant</center></h1>"
 PLACEHOLDER = "Ask me anything! I'll think through it step by step."
 
 CSS = """
-
-duplicate-button {
+.duplicate-button {
     margin: auto !important;
     color: white !important;
     background: black !important;
@@ -84,43 +81,56 @@ h3 {
     height: 500px !important;
     overflow-y: auto !important;
 }
-
 """
 
-# Precompile regex patterns for better performance
-TAG_PATTERNS = [
-    (re.compile(r'<Thinking>'), '\n<Thinking>\n'),
-    (re.compile(r'</Thinking>'), '\n</Thinking>\n'),
-    (re.compile(r'<Critique>'), '\n<Critique>\n'),
-    (re.compile(r'</Critique>'), '\n</Critique>\n'),
-    (re.compile(r'<Revising>'), '\n<Revising>\n'),
-    (re.compile(r'</Revising>'), '\n</Revising>\n'),
-    (re.compile(r'<Final>'), '\n<Final>\n'),
-    (re.compile(r'</Final>'), '\n</Final>\n')
-]
-
 def format_text(text: str) -> str:
     """Format text with proper spacing and tag highlighting"""
+    tag_patterns = [
+        (r'<Thinking>', '\n<Thinking>\n'),
+        (r'</Thinking>', '\n</Thinking>\n'),
+        (r'<Critique>', '\n<Critique>\n'),
+        (r'</Critique>', '\n</Critique>\n'),
+        (r'<Revising>', '\n<Revising>\n'),
+        (r'</Revising>', '\n</Revising>\n'),
+        (r'<Final>', '\n<Final>\n'),
+        (r'</Final>', '\n</Final>\n')
+    ]
+    
     formatted = text
-    for pattern, replacement in TAG_PATTERNS:
-        formatted = pattern.sub(replacement, formatted)
-    return '\n'.join(line for line in formatted.split('\n') if line.strip())
+    for pattern, replacement in tag_patterns:
+        formatted = re.sub(pattern, replacement, formatted)
+    
+    formatted = '\n'.join(line for line in formatted.split('\n') if line.strip())
+    
+    return formatted
 
 def format_chat_history(history: List[List[str]]) -> str:
     """Format chat history for display"""
-    return "\n\n".join(
-        f"User: {user_msg}\nAssistant: {assistant_msg}" 
-        for user_msg, assistant_msg in history if user_msg
-    )
+    formatted = []
+    for user_msg, assistant_msg in history:
+        formatted.append(f"User: {user_msg}")
+        if assistant_msg:
+            formatted.append(f"Assistant: {assistant_msg}")
+    return "\n\n".join(formatted)
+
+def create_examples() -> List[str]:
+    """Create example queries for the UI"""
+    return [
+        "Explain the concept of artificial intelligence.",
+        "How does photosynthesis work?",
+        "What are the main causes of climate change?",
+        "Describe the process of protein synthesis.",
+        "What are the key features of a democratic government?",
+    ]
 
 def convert_history_to_cohere_format(history: List[List[str]]) -> List[dict]:
-    """Convert chat history to Cohere's format with proper roles"""
+    """Convert chat history to Cohere's format"""
     cohere_history = []
     for user_msg, assistant_msg in history:
         if user_msg:
-            cohere_history.append({"role": "USER", "message": user_msg})
+            cohere_history.append({"role": "User", "message": user_msg})
         if assistant_msg:
-            cohere_history.append({"role": "CHATBOT", "message": assistant_msg})
+            cohere_history.append({"role": "Chatbot", "message": assistant_msg})
     return cohere_history
 
 def chat_response(
@@ -135,42 +145,45 @@ def chat_response(
     penalty: float = 1.2,
     api_key: str = os.getenv("COHERE_API_KEY")
 ) -> Iterator[Tuple[List[List[str]], str]]:
-    """Generate chat responses using Cohere API with error handling"""
-    if not api_key:
-        error_message = "Error: COHERE_API_KEY environment variable not set."
-        history.append([message, error_message])
-        yield history, format_chat_history(history)
-        return
-
+    """Generate chat responses using Cohere API"""
     co = cohere.Client(api_key=api_key)
+    
+    # Convert history to Cohere format
     chat_history = convert_history_to_cohere_format(history)
+    
+    # Initialize buffer for streaming
     buffer = ""
-    history.append([message, ""])
+    history = history + [[message, ""]]
     
+    # Process stream
     try:
+        # Initialize stream
         stream = co.chat_stream(
             model='c4ai-aya-expanse-32b',
             message=message,
             temperature=temperature,
-            max_tokens=max_new_tokens,
-            p=top_p,
-            k=top_k,
-            frequency_penalty=penalty,
             chat_history=chat_history,
             prompt_truncation='AUTO',
             preamble=system_prompt
         )
-
+        
         for event in stream:
             if event.event_type == "text-generation":
                 buffer += event.text
-                history[-1][1] = format_text(buffer)
-                yield history, format_chat_history(history)
-
+                formatted_buffer = format_text(buffer)
+                history[-1][1] = formatted_buffer
+                chat_display = format_chat_history(history)
+                yield history, chat_display
+                
     except Exception as e:
         error_message = f"Error: {str(e)}"
         history[-1][1] = error_message
-        yield history, format_chat_history(history)
+        chat_display = format_chat_history(history)
+        yield history, chat_display
+
+def process_example(example: str) -> tuple:
+    """Process example query and return empty history and updated display"""
+    return [], f"User: {example}\n\n"
 
 def main():
     """Main function to set up and launch the Gradio interface"""
@@ -187,19 +200,18 @@ def main():
                 chat_display = gr.TextArea(
                     value="",
                     label="Chat History",
-                    interactive=False,
+                    interactive=AttributeError,
                     elem_classes=["chat-area"],
                 )
                 
                 message = gr.TextArea(
                     placeholder=PLACEHOLDER,
                     label="Your message",
-                    lines=3,
-                    elem_id="message_input"
+                    lines=5
                 )
                 
                 with gr.Row():
-                    submit = gr.Button("Send", elem_id="submit_btn")
+                    submit = gr.Button("Send")
                     clear = gr.Button("Clear")
                 
                 with gr.Accordion("⚙️ Advanced Settings", open=False):
@@ -243,32 +255,62 @@ def main():
                         value=1.2,
                         label="Repetition Penalty",
                     )
-
-        # Event handlers
-        submit.click(
+                
+                examples = gr.Examples(
+                    examples=create_examples(),
+                    inputs=[message],
+                    outputs=[chat_history, chat_display],
+                    fn=process_example,
+                    cache_examples=False,
+                )
+        
+        # Set up event handlers
+        submit_click = submit.click(
             chat_response,
-            inputs=[message, chat_history, chat_display, system_prompt,
-                    temperature, max_tokens, top_p, top_k, penalty],
+            inputs=[
+                message,
+                chat_history,
+                chat_display,
+                system_prompt,
+                temperature,
+                max_tokens,
+                top_p,
+                top_k,
+                penalty,
+            ],
             outputs=[chat_history, chat_display],
             show_progress=True,
-        ).then(lambda: "", outputs=message)
-
+        )
+        
+        # Bind Enter key to submit
         message.submit(
             chat_response,
-            inputs=[message, chat_history, chat_display, system_prompt,
-                    temperature, max_tokens, top_p, top_k, penalty],
+            inputs=[
+                message,
+                chat_history,
+                chat_display,
+                system_prompt,
+                temperature,
+                max_tokens,
+                top_p,
+                top_k,
+                penalty,
+            ],
             outputs=[chat_history, chat_display],
             show_progress=True,
-        ).then(lambda: "", outputs=message)
-
+        )
+        
         clear.click(
             lambda: ([], ""),
             outputs=[chat_history, chat_display],
             show_progress=True,
         )
-
-    # Add JavaScript for Enter key handling
+        
+        # Clear input after submission
+        submit_click.then(lambda: "", outputs=message)
+        message.submit(lambda: "", outputs=message)
     
+    return demo
 
 if __name__ == "__main__":
     demo = main()