Update app.py

app.py

@@ -11,7 +11,6 @@ from docx import Document
 from docx.enum.text import WD_PARAGRAPH_ALIGNMENT
 from io import BytesIO
 
-
 # Set up OpenAI API
 openai.api_key = "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"
 
@@ -19,14 +18,10 @@ openai.api_key = "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"
 st.markdown(
     """
 <style>
-    body {
-        background-color: transparent;
-    }
     .container {
         display: flex;
         justify-content: center;
         align-items: center;
-        background-color: rgba(255, 255, 255, 0.7);
         border-radius: 15px;
         padding: 20px;
     }
@@ -42,14 +37,12 @@ model = CLIPModel().to(device)
 model.load_state_dict(torch.load("weights.pt", map_location=torch.device('cpu')))
 text_embeddings = torch.load('saved_text_embeddings.pt', map_location=device)
 
-
 def show_predicted_caption(image):
     matches = predict_caption(
         image, model, text_embeddings, testing_df["caption"]
     )[0]
     return matches
 
-
 def generate_radiology_report(prompt):
     response = openai.Completion.create(
         engine="text-davinci-003",
@@ -61,7 +54,6 @@ def generate_radiology_report(prompt):
     )
     return response.choices[0].text.strip()
 
-
 def chatbot_response(prompt):
     response = openai.Completion.create(
         engine="text-davinci-003",
@@ -73,7 +65,6 @@ def chatbot_response(prompt):
     )
     return response.choices[0].text.strip()
 
-# Add this function to your code
 def save_as_docx(text, filename):
     document = Document()
     document.add_paragraph(text)
@@ -82,13 +73,11 @@ def save_as_docx(text, filename):
         output.seek(0)
         return output.getvalue()
 
-# Add this function to your code
 def download_link(content, filename, link_text):
     b64 = base64.b64encode(content).decode()
     href = f'<a href="data:application/octet-stream;base64,{b64}" download="{filename}">{link_text}</a>'
     return href
 
-
 st.title("RadiXGPT: An Evolution of machine doctors towrads Radiology")
 st.write("Upload Scan to get Radiological Report:")
 
@@ -104,7 +93,6 @@ if uploaded_file is not None:
             caption = show_predicted_caption(image_np)
             st.success(f"Caption: {caption}")
 
-            # Add the OpenAI API call here and generate the radiology report
             radiology_report = generate_radiology_report(f"Write Complete Radiology Report for this: {caption}")
             container = st.container()
             with container:
@@ -117,10 +105,13 @@ if uploaded_file is not None:
             st.write("Ask any questions you have about the radiology report:")
 
             user_input = st.text_input("Enter your question:")
-            if user_input:
-                if user_input.lower() == "thank you":
-                    st.write("Bot: You're welcome! If you have any more questions, feel free to ask.")
-                else:
-                    # Add the OpenAI API call here and generate the answer to the user's question
+            while user_input.lower() != "thank you":
+                if user_input:
                     answer = chatbot_response(f"Answer to the user's question based on the generated radiology report: {user_input}")
-                    st.write(f"Bot: {answer}")
+                    st.write(f"Bot: {answer}")
+                    user_input = st.text_input("Enter your question:")
+                else:
+                    break
+
+            if user_input.lower() == "thank you":
+                st.write("Bot: You're welcome! If you have any more questions, feel free to ask.")