Update interim.py

interim.py

@@ -2,7 +2,6 @@ import os
 from datetime import datetime
 import streamlit as st
 from patentwiz import preprocess_data, qa_agent
-from patentwiz.main import PROMPT
 
 # Check if the API key is loaded
 api_key = os.getenv("OPENAI_API_KEY")
@@ -10,6 +9,40 @@ if not api_key:
     st.error("OPENAI_API_KEY not found! Please set it in the environment variables or Hugging Face Secrets.")
     st.stop()
 
+PROMPT = """
+Task: Carefully review the given patent text and extract as much physical measurements information such as length/distance, mass/weight, time, temperature, Volume, area, speed, pressure, energy, power, electric current 
+and voltage, frequency, force, acceleration, density, resistivity, magnetic field strength, and luminous intensity as much as possible. 
+We are particularly interested in physical measurements including substance that was measured, Value of the measurement, and Unit of the measurement, and measurement type mentioned in the text. 
+For each measurement, please provide the following details:
+- The substance that was measured. (substance)
+- The specific value or range that was measured. (Measured Value)
+- The unit of the measurement, if provided. (Unit)
+- The type of measurement being conducted (e.g., diameter, size, etc.) 
+Format your response in a structured JSON-like format, as follows:
+{"Content": [
+    {
+      "Measurement_substance": "substance",
+      "Measured_value": "value",
+      "Measured_unit": "unit",
+      "measurement_type": "type"
+    },
+    // ... additional measurements, if present
+  ]
+}
+If multiple measurements are present in the text, each should be listed as a separate object within the "Content" array.
+Example: If the text includes the sentence, "The resulting BaCO3 had a crystallite size of between about 20 and 40 nm", the output should be:
+{"Content": [
+    {
+      "Measurement_substance": "BaCO3",
+      "Measured_value": "between about 20 and 40",
+      "Measured_unit": "nm",
+      "measurement_type": "crystallite size"
+    }
+  ]
+}
+Try to provide as complete and accurate information as possible. Print only the formatted JSON response.
+"""
+
 # Title and description
 st.title("Technical Measurements Extractor for Patents")
 st.write(
@@ -19,7 +52,7 @@ st.write(
 
 # User Input Section
 st.header("Enter Details for Patent Analysis")
-user_date_input = st.text_input("Enter a date in the format 'YYYY-MM-DD':", help="e.g., 2024-01-01")
+user_date_input = st.text_input("Enter a date in the format 'YYYY-MM-DD':", value="2024-06-16")
 
 num_patents_to_analyze = st.number_input(
     "Number of patents to analyze:", min_value=1, value=1, step=1, help="Specify how many patents you want to analyze."