Create app.py

app.py

@@ -0,0 +1,53 @@
+import streamlit as st
+from openai import OpenAI
+
+# Set your API key and base URL
+API_KEY = os.getenv("API_KEY")
+BASE_URL = "https://openrouter.ai/api/v1"
+
+# Initialize the OpenAI client with OpenRouter settings
+client = OpenAI(
+    base_url=BASE_URL,
+    api_key=API_KEY,
+)
+
+st.title("Chat with OpenRouter API")
+
+# Initialize session state for storing the conversation if it doesn't exist
+if 'messages' not in st.session_state:
+    st.session_state.messages = [
+        {"role": "assistant", "content": "Hello! How can I help you today?"}
+    ]
+
+# Display the conversation
+for msg in st.session_state.messages:
+    if msg["role"] == "user":
+        st.markdown(f"**User:** {msg['content']}")
+    else:
+        st.markdown(f"**Assistant:** {msg['content']}")
+
+# Input field for user to type a new message
+user_input = st.text_input("Your message:", key="input")
+
+if st.button("Send") and user_input:
+    # Append the user's message to the conversation
+    st.session_state.messages.append({"role": "user", "content": user_input})
+    
+    # Call the OpenRouter API using the chosen (different) model
+    try:
+        response = client.chat.completions.create(
+            extra_headers={
+                "HTTP-Referer": "http://your-site-url.com",  # Optional
+                "X-Title": "Your Site Title"  # Optional
+            },
+            extra_body={},
+            model="deepseek/deepseek-v3-large:free",  # Changed model
+            messages=st.session_state.messages,
+        )
+        assistant_reply = response.choices[0].message.content
+    except Exception as e:
+        assistant_reply = f"An error occurred: {str(e)}"
+    
+    # Append the assistant's reply and refresh the UI
+    st.session_state.messages.append({"role": "assistant", "content": assistant_reply})
+    st.experimental_rerun()