Update app.py

app.py

@@ -29,46 +29,127 @@ GREETING_MESSAGES = [
     "The universe awaits! I'm AstroSage. What astronomical wonders shall we discuss?",
 ]
 
-def respond_stream(message, history):
-    if not message:
-        return
-
-    system_message = "Assume the role of AstroSage, a helpful chatbot designed to answer user queries about astronomy, astrophysics, and cosmology."
-    messages = [{"role": "system", "content": system_message}]
-    for user, assistant in history:
-        messages.append({"role": "user", "content": user})
+def format_chat_history(history):
+    """Format the chat history for the model."""
+    formatted_messages = []
+    for human, assistant in history:
+        formatted_messages.append({"role": "user", "content": human})
         if assistant:
-            messages.append({"role": "assistant", "content": assistant})
-    messages.append({"role": "user", "content": message})
+            formatted_messages.append({"role": "assistant", "content": assistant})
+    return formatted_messages
 
-    try:
-        past_tokens = ""  # Accumulate and yield all tokens so far
-        for chunk in llm.create_chat_completion(
-            messages=messages,
-            max_tokens=512,
-            temperature=0.7,
-            top_p=0.9,
-            stream=True
-        ):
-            delta = chunk["choices"][0]["delta"]
-            if "content" in delta:
-                new_tokens = delta["content"]
-                past_tokens += new_tokens
-                yield past_tokens  # Yield the accumulated response to allow streaming
-    except Exception as e:
-        yield f"Error during generation: {e}"
+def generate_response(message, history):
+    """Generate response using the LLM."""
+    if not message:
+        return random.choice(GREETING_MESSAGES)
+    
+    formatted_history = format_chat_history(history)
+    
+    response = llm.create_chat_completion(
+        messages=[
+            {"role": "system", "content": "You are AstroSage, an intelligent AI assistant specializing in astronomy, astrophysics, and space science. You provide accurate, scientific information while making complex concepts accessible. You're enthusiastic about space exploration and maintain a sense of wonder about the cosmos."},
+            *formatted_history,
+            {"role": "user", "content": message}
+        ],
+        max_tokens=512,
+        temperature=0.7,
+        top_p=0.95,
+        stream=True
+    )
+    
+    partial_message = ""
+    for chunk in response:
+        if chunk and "content" in chunk["choices"][0]["delta"]:
+            partial_message += chunk["choices"][0]["delta"]["content"]
+            yield partial_message
 
-initial_message = random.choice(GREETING_MESSAGES)
-chatbot = gr.Chatbot([[None, initial_message]]).style(height=750)  # Set height
+# Custom CSS for a space theme
+custom_css = """
+#component-0 {
+    background-color: #1a1a2e;
+    border-radius: 15px;
+    padding: 20px;
+}
+.dark {
+    background-color: #0f0f1a;
+}
+.contain {
+    max-width: 1200px !important;
+}
+"""
 
-with gr.Blocks() as demo:
+# Create the Gradio interface
+with gr.Blocks(css=custom_css, theme=gr.themes.Soft(primary_hue="indigo", neutral_hue="slate")) as demo:
+    gr.Markdown(
+        """
+        # 🌌 AstroSage: Your Cosmic AI Companion
+        
+        Welcome to AstroSage, an advanced AI assistant specializing in astronomy, astrophysics, and space science. 
+        Powered by the AstroSage-8B model, I'm here to help you explore the wonders of the universe!
+        
+        ### What Can I Help You With?
+        - 🪐 Explanations of astronomical phenomena
+        - 🚀 Space exploration and missions
+        - ⭐ Stars, galaxies, and cosmic objects
+        - 🌍 Planetary science and exoplanets
+        - 📊 Astrophysics concepts and theories
+        - 🔭 Astronomical instruments and observations
+        
+        Just type your question below and let's embark on a cosmic journey together!
+        """
+    )
+    
+    chatbot = gr.Chatbot(
+        label="Chat with AstroSage",
+        bubble_full_width=False,
+        show_label=True,
+        height=450
+    )
+    
     with gr.Row():
-        with gr.Column(scale=0.8):
-            chatbot.render()
-
-        with gr.Column(scale=0.2):
-           clear = gr.Button("Clear")
-
-    clear.click(lambda: [], None, chatbot,queue=False)
+        msg = gr.Textbox(
+            label="Type your message here",
+            placeholder="Ask me anything about space and astronomy...",
+            scale=9
+        )
+        clear = gr.Button("Clear Chat", scale=1)
+    
+    # Example questions for quick start
+    gr.Examples(
+        examples=[
+            "What is a black hole and how does it form?",
+            "Can you explain the life cycle of a star?",
+            "What are exoplanets and how do we detect them?",
+            "Tell me about the James Webb Space Telescope.",
+            "What is dark matter and why is it important?"
+        ],
+        inputs=msg,
+        label="Example Questions"
+    )
+    
+    # Event handlers
+    msg.submit(
+        generate_response,
+        [msg, chatbot],
+        [msg, chatbot],
+        queue=True
+    )
+    
+    clear.click(
+        lambda: (None, None),
+        None,
+        [msg, chatbot],
+        queue=False
+    )
+    
+    # Initial greeting
+    demo.load(
+        lambda: (random.choice(GREETING_MESSAGES), []),
+        None,
+        [msg, chatbot],
+        queue=False
+    )
 
-    demo.queue().launch()
+# Launch the app
+if __name__ == "__main__":
+    demo.launch()