Update app.py

app.py

@@ -8,15 +8,36 @@ from langchain_community.llms import HuggingFaceHub
 from langchain.prompts import ChatPromptTemplate
 from dotenv import load_dotenv
 import os
+import numpy as np
 from datetime import datetime
 from skyfield.api import load
 import matplotlib.pyplot as plt
 from io import BytesIO
 from PIL import Image
+from opentelemetry.sdk.trace import TracerProvider
+from openinference.instrumentation.smolagents import SmolagentsInstrumentor
+from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter
+from opentelemetry.sdk.trace.export import SimpleSpanProcessor
 
 # Load environment variables
 load_dotenv()
 
+# Add the alias before instrumentation
+smolagents.ApiModel = smolagents.HfApiModel
+
+LANGFUSE_PUBLIC_KEY="pk-lf-133099c7-8644-49e8-8f6e-ec8bd6d543fd"
+LF_SECRET_KEY = os.environ["LANGFUSE_SECRET_KEY"]
+LANGFUSE_AUTH=base64.b64encode(f"{LANGFUSE_PUBLIC_KEY}:{LF_SECRET_KEY}".encode()).decode()
+
+os.environ["OTEL_EXPORTER_OTLP_ENDPOINT"] = "https://cloud.langfuse.com/api/public/otel" # EU data region
+# os.environ["OTEL_EXPORTER_OTLP_ENDPOINT"] = "https://us.cloud.langfuse.com/api/public/otel" # US data region
+os.environ["OTEL_EXPORTER_OTLP_HEADERS"] = f"Authorization=Basic {LANGFUSE_AUTH}"
+
+trace_provider = TracerProvider()
+trace_provider.add_span_processor(SimpleSpanProcessor(OTLPSpanExporter()))
+
+SmolagentsInstrumentor().instrument(tracer_provider=trace_provider)
+
 DATA_PATH = ""  # Specify the path to your file
 PROMPT_TEMPLATE = """
 Ответь на вопрос, используя только следующий контекст:
@@ -136,6 +157,25 @@ def process_query(query_text: str, vectorstore):
     except Exception as e:
         return f"Ошибка обработки запроса: {str(e)}", []
 
+# Function to parse date and time into ISO format
+def parse_date_time(date_time_str):
+    try:
+        dt = parser.parse(date_time_str)
+        return dt.isoformat()
+    except ValueError:
+        return None
+
+# Function to convert longitude to zodiac sign and degrees
+def lon_to_sign(lon):
+    signs = ["Овен", "Телец", "Близнецы", "Рак", "Лев", "Дева", 
+             "Весы", "Скорпион", "Стрелец", "Козерог", "Водолей", "Рыбы"]
+    sign_index = int(lon // 30)
+    sign = signs[sign_index]
+    degrees = int(lon % 30)
+    minutes = int((lon % 1) * 60)
+    return f"{sign} {degrees}°{minutes}'"
+
+# Function to calculate PLadder and zone sizes
 def PLadder_ZSizes(date_time_iso: str):
     """
     Calculate the planetary ladder and zone sizes for a given date and time.
@@ -238,22 +278,22 @@ def plot_pladder(PLadder):
         matplotlib.figure.Figure: The generated plot
     """
     fig, ax = plt.subplots()
-    # Draw triangle with vertices (0,0), (0,3), (3,0)
-    ax.plot([0, 0, 3, 0], [0, 3, 0, 0], 'k-')
+    # Plot triangle with right angle on top: vertices at (0,0), (1.5,3), (3,0)
+    ax.plot([0, 1.5, 3, 0], [0, 3, 0, 0], 'k-')
     # Draw horizontal lines dividing height into three equal parts
     ax.plot([0, 3], [1, 1], 'k--')
     ax.plot([0, 3], [2, 2], 'k--')
-    # Define positions for planets 1 to 7
-    positions = [(0, 0), (0, 1), (0, 2), (0, 3), (1, 2), (2, 1), (3, 0)]
+    # Define positions for planets 1 to 7, adjusted to avoid overlap
+    positions = [(0.2, 0.2), (0.2, 1.2), (0.2, 2.2), (1.5, 3.2), (2.8, 2.2), (2.8, 1.2), (2.8, 0.2)]
     for i, pos in enumerate(positions):
         symbol = planet_symbols[PLadder[i]]
-        ax.text(pos[0], pos[1], symbol, ha='center', va='center', fontsize=12)
+        ax.text(pos[0], pos[1], symbol, ha='center', va='center', fontsize=24)  # Doubled font size
     ax.set_xlim(-0.5, 3.5)
     ax.set_ylim(-0.5, 3.5)
     ax.set_aspect('equal')
     ax.axis('off')
     return fig
-
+    
 def chat_interface(query_text):
     """
     Handle user queries, either for planetary ladder or general RAG questions.
@@ -323,14 +363,31 @@ def chat_interface(query_text):
         return f"Критическая ошибка: {str(e)}", None
 
 # Define Gradio Interface
-interface = gr.Interface(
-    fn=chat_interface,
-    inputs=gr.Textbox(lines=2, placeholder="Введите ваш вопрос здесь..."),
-    outputs=[gr.Textbox(), gr.Image()],
-    title="Чат с документами",
-    description="Задайте вопрос, и я отвечу на основе загруженных документов. "
-                "Для запроса планетарной лестницы используйте формат: PLadder YYYY-MM-DDTHH:MM:SS"
-)
+#interface = gr.Interface(
+#    fn=chat_interface,
+#    inputs=gr.Textbox(lines=2, placeholder="Введите ваш вопрос здесь..."),
+#    outputs=[gr.Textbox(), gr.Image()],
+#    title="Чат с документами",
+#    description="Задайте вопрос, и я отвечу на основе книги Павла Глобы "Планетарная Лестница". "
+#                "Для быстрого запроса трактовки планетарной лестницы используйте формат: PLadder DD-MM-YYYY HH:MM:SS место"
+#)
+
+# UI layout with Gradio Blocks
+with gr.Blocks() as interface:
+    with gr.Row():
+        with gr.Column(scale=2):
+            output_text = gr.Textbox(label="Response", lines=10)
+        with gr.Column(scale=1):
+            output_image = gr.Image(label="Planetary Ladder Plot")
+    with gr.Row():
+        query_text = gr.Textbox(label="Query", placeholder="e.g., PLadder 2023-10-10 12:00")
+        location_lat = gr.Textbox(label="Latitude", placeholder="e.g., 37.7749")
+        location_lon = gr.Textbox(label="Longitude", placeholder="e.g., -122.4194")
+    
+    query_text.submit(chat_interface, 
+                      inputs=[query_text, location_lat, location_lon], 
+                      outputs=[output_text, output_image])
+
 
 if __name__ == "__main__":
     interface.launch()