Create app.py

app.py

@@ -0,0 +1,66 @@
+import gradio as gr
+from huggingface_hub import InferenceClient
+import os
+from typing import Generator
+
+# Initialize Inference API client
+# Try AstroSage-Llama-3.1-8B; fallback to Llama-3.2-3B-Instruct if unsupported
+model = "AstroMLab/AstroSage-Llama-3.1-8B"  # Replace with "meta-llama/Llama-3.2-3B-Instruct" if needed
+client = InferenceClient(model=model, token=os.getenv("HF_TOKEN"))
+
+def generate_astronomy_explanation(prompt: str) -> Generator[str, None, None]:
+    """
+    Generates astronomy-related explanations using the model via Inference API with streaming.
+    """
+    system_prompt = (
+        "You are AstroSage, an expert astronomy assistant like Grok. Given a user prompt about "
+        "astronomical phenomena, space exploration, stars, galaxies, cosmology, planetary science, "
+        "or astrophysics, provide a clear, educational explanation. Stream the output line by line. "
+        "Use bullet points for key points and keep responses concise yet informative."
+    )
+    messages = [
+        {"role": "system", "content": system_prompt},
+        {"role": "user", "content": prompt}
+    ]
+
+    # Stream output from Inference API using chat_completion
+    for chunk in client.chat_completion(
+        messages=messages,
+        max_tokens=1000,  # Balanced for complete answers and speed
+        temperature=0.7,
+        top_p=0.9,
+        stream=True
+    ):
+        # Extract content from the chunk
+        content = chunk.choices[0].delta.content
+        if content:  # Only yield non-empty content
+            yield content
+
+# Gradio interface with streaming
+def live_astronomy_generator(prompt: str):
+    """
+    Handles streaming output for the Gradio interface.
+    """
+    output = ""
+    for chunk in generate_astronomy_explanation(prompt):
+        output += chunk
+        yield output
+
+# Gradio app
+with gr.Blocks() as demo:
+    gr.Markdown("# Invescoz AI Studio: Astronomy Explainer")
+    prompt_input = gr.Textbox(
+        label="Enter your astronomy question",
+        placeholder="e.g., Explain how black holes form or Describe the lifecycle of a star"
+    )
+    output_display = gr.Textbox(label="Astronomy Explanation", interactive=False, lines=10)
+    submit_button = gr.Button("Generate")
+    
+    submit_button.click(
+        fn=live_astronomy_generator,
+        inputs=prompt_input,
+        outputs=output_display
+    )
+
+# Launch the app (handled by Hugging Face Spaces)
+demo.launch()