Update app.py

app.py

@@ -92,20 +92,48 @@ with left:
     st.text_input("Calculation", st.session_state.calc_input, key="display", disabled=True)
 
 with lr:
-    # End of Box 1 and first Carousel Item
-    st.markdown('''<h3><i class="fa fa-pencil"></i> Form 1</h3>''', unsafe_allow_html=True)
 
-    # Box 2: Form 1
-    persona = st.text_input("Input a persona, e.g. photographer", value="photographer")
+    st.markdown('''<h3><i class="fa fa-image"></i> Gen Image</h3>''', unsafe_allow_html=True)
+
+    api_key = f"{BASETEN_KEY}"
+    negative_prompt = st.text_input("Negative Prompt", "blurry, text, low quality")
+    positive_prompt = st.text_input("Positive Prompt", "An igloo on a snowy day, 4k, hd")
+    controlnet_image_url = st.text_input("ControlNet Image URL", "https://storage.googleapis.com/logos-bucket-01/baseten_logo.png")
+
+
 
     # Button to trigger generation
     if st.button("Generate Prompt"):
-        if persona:
-            with st.spinner("Generating..."):
-                result = "Test"
-            st.text_area("Generated Prompt", value=result, height=200)
+       # Making the API request
+        response = requests.post(
+            "https://model-7wlx9oew.api.baseten.co/production/predict",
+            headers={"Authorization": f"Api-Key {api_key}"},
+            json={
+                'workflow_values': {
+                    'negative_prompt': negative_prompt,
+                    'positive_prompt': positive_prompt,
+                    'controlnet_image': controlnet_image_url
+                }
+            }
+        )
+        
+        # Display the response
+        if response.status_code == 200:
+            result = response.json().get("result")
+            if result:
+                image_data = result[0].get("data")
+                if image_data:
+                    # Decode the base64 image data
+                    image = base64.b64decode(image_data)
+                    # Display the image in Streamlit
+                    st.image(image, caption="Generated Image", use_column_width=True)
+                else:
+                    st.error("No image data found in the response.")
+            else:
+                st.error("No result found in the response.")
         else:
-            st.error("Please enter a persona to generate a prompt.")
+            st.error(f"Error: {response.status_code}, {response.text}")
+
 
 with rl:
     # End of Box 2 and second Carousel Item
@@ -131,16 +159,16 @@ with rl:
             with st.spinner("Transcribing..."):
                 #transcription = transcribe("temp_recording.wav")
                 #need to send the data here
+                transcription = "Under Process"
                 print("")
             st.text_area("Transcription", transcription, height=200)
         else:
-            st.error("Please record audio or upload a file to transcribe.")
+            st.error("Please record audio or upload a file to transcribe.")    
+with right:
 
-    # End of Box 3 and third Carousel Item
     st.markdown('''<h3><i class="fa fa-pencil"></i> Chat with Mistral</h3>''', unsafe_allow_html=True)
-with right:
     # Box 4: Form 3
-    prompt3 = st.text_input("Enter Prompt", key="prompt3")
+    prompt3 = st.text_input("Enter Prompt", key="prompt3", value="Why is Sky Blue?")
     #image_url3 = st.text_input("Enter Image URL", key="image_url3")
     if st.button("Submit", key="submit3"):
         payload = {"prompt": prompt3}