Reverted Updates due to unexpected errors.

app.py

@@ -2,66 +2,45 @@ import streamlit as st
 from openai import OpenAI
 import os
 
-# Set up NVIDIA API client
+# Set your API key here
 client = OpenAI(
     base_url="https://integrate.api.nvidia.com/v1",
     api_key=os.environ.get("NVIDIA_API_KEY")
 )
 
-def query_ai_model(prompt, model="meta/llama-3.1-405b-instruct", temperature=0.7, max_tokens=512, top_p=0.9):
+def query_ai_model(prompt):
     try:
+        # Example of using NVIDIA's OpenAI integration
         completion = client.chat.completions.create(
-            model=model,
+            model="meta/llama-3.1-405b-instruct",
             messages=[{"role": "user", "content": prompt}],
-            temperature=temperature,
-            top_p=top_p,
-            max_tokens=max_tokens,
+            temperature=0.2,
+            top_p=0.7,
+            max_tokens=1024,
             stream=True
         )
         response = ""
         response_container = st.empty()
-        
+
         for chunk in completion:
             if chunk.choices[0].delta.content is not None:
                 response += chunk.choices[0].delta.content
                 response_container.markdown(response)
     except Exception as e:
-        st.error(f"An error occurred: {str(e)}")
+        return f"An error occurred: {str(e)}"
 
-# Streamlit UI
+# Streamlit App
 st.title("Mark's AI Chatbot")
-<<<<<<< HEAD
-st.write("Provide a topic and customize the response criteria.")
-=======
-st.write("Enter a prompt below and click 'Generate Answer' to get a response from the AI model.")
->>>>>>> 16dd513590026069c27327e4c9dd4e29445fccaf
+st.write("Enter a prompt below and click 'Ask AI' to get a response from the AI model.")
 
-# Input Fields
+# Input prompt
 user_input = st.text_area("Your Prompt:", placeholder="Type something...")
 
-# Dropdown Menus
-output_format = st.selectbox("Select Output Format:", ["Story", "Poem", "Article", "Code"])
-tone_style = st.selectbox("Select Tone/Style:", ["Formal", "Informal", "Humorous", "Technical"])
-
-# Sliders/Numeric Inputs
-creativity_level = st.slider("Creativity Level:", min_value=0.0, max_value=1.0, value=0.7, step=0.1)
-max_length = st.slider("Max Length (tokens):", min_value=100, max_value=1024, value=512, step=50)
-num_responses = st.number_input("Number of Responses:", min_value=1, max_value=5, value=1, step=1)
-
-# Checkboxes
-enable_creativity = st.checkbox("Enable Creative Mode", value=True)
-fact_checking = st.checkbox("Enable Fact-Checking")
-
 if st.button("Generate Answer"):
     if user_input.strip():
-        with st.spinner("Generating response..."):
-            full_prompt = f"Format: {output_format}\nTone: {tone_style}\nPrompt: {user_input}"
-            ai_response = query_ai_model(
-                full_prompt,
-                temperature=creativity_level if enable_creativity else 0.2,
-                max_tokens=max_length
-            )
+        with st.spinner("Getting a response from the AI model..."):
+            ai_response = query_ai_model(user_input)
         st.success("AI Response:")
         st.write(ai_response)
     else:
-        st.warning("Please enter a prompt before clicking the button.")
+        st.warning("Please enter a prompt before clicking the Button.")